Google Git
Sign in
android / platform / external / pytorch / 96eec95ece / . / caffe2 / operators
tree: 450d85efce34592d3f7fbad258a488a2cb7e982d [path history] [tgz]
  1. experimental/
  2. hip/
  3. quantized/
  4. rnn/
  5. abs_op.cc
  6. abs_op.cu
  7. abs_op.h
  8. accumulate_op.cc
  9. accumulate_op.cu
  10. accumulate_op.h
  11. accuracy_op.cc
  12. accuracy_op.cu
  13. accuracy_op.h
  14. acos_op.cc
  15. acos_op.cu
  16. acos_op.h
  17. activation_ops_cudnn.h
  18. affine_channel_op.cc
  19. affine_channel_op.cu
  20. affine_channel_op.h
  21. alias_with_name.cc
  22. alias_with_name.cu
  23. alias_with_name.h
  24. apmeter_op.cc
  25. apmeter_op.h
  26. arg_ops.cc
  27. arg_ops.cu
  28. arg_ops.h
  29. asin_op.cc
  30. asin_op.cu
  31. asin_op.h
  32. assert_op.cc
  33. assert_op.cu
  34. assert_op.h
  35. atan_op.cc
  36. atan_op.cu
  37. atan_op.h
  38. atomic_ops.cc
  39. batch_box_cox_op.cc
  40. batch_box_cox_op.h
  41. batch_bucketize_op.cc
  42. batch_bucketize_op.h
  43. batch_gather_ops.cc
  44. batch_gather_ops.cu
  45. batch_gather_ops.h
  46. batch_matmul_op.cc
  47. batch_matmul_op.cu
  48. batch_matmul_op.h
  49. batch_matmul_op_gpu_test.cc
  50. batch_matmul_op_test.cc
  51. batch_moments_op.cc
  52. batch_moments_op.cu
  53. batch_moments_op.h
  54. batch_permutation_op.cc
  55. batch_permutation_op.cu
  56. batch_permutation_op.h
  57. batch_permutation_op_gpu_test.cc
  58. batch_sparse_to_dense_op.cc
  59. batch_sparse_to_dense_op.cu
  60. batch_sparse_to_dense_op.h
  61. bbox_transform_op.cc
  62. bbox_transform_op.h
  63. bisect_percentile_op.cc
  64. bisect_percentile_op.h
  65. boolean_mask_ops.cc
  66. boolean_mask_ops.cu
  67. boolean_mask_ops.h
  68. boolean_unmask_ops.cc
  69. boolean_unmask_ops.cu
  70. boolean_unmask_ops.h
  71. boolean_unmask_ops_test.cc
  72. box_with_nms_limit_op.cc
  73. box_with_nms_limit_op.h
  74. bucketize_op.cc
  75. bucketize_op.cu
  76. bucketize_op.h
  77. byte_weight_dequant_op.cc
  78. byte_weight_dequant_op.h
  79. cast_op.cc
  80. cast_op.cu
  81. cast_op.h
  82. cbrt_op.cc
  83. cbrt_op.cu
  84. cbrt_op.h
  85. cc_bmm_bg_op.cc
  86. cc_bmm_bg_op.h
  87. ceil_op.cc
  88. ceil_op.cu
  89. ceil_op.h
  90. channel_backprop_stats_op.cc
  91. channel_backprop_stats_op.cu
  92. channel_backprop_stats_op.h
  93. channel_shuffle_op.cc
  94. channel_shuffle_op.cu
  95. channel_shuffle_op.h
  96. channel_stats_op.cc
  97. channel_stats_op.cu
  98. channel_stats_op.h
  99. channelwise_conv3d_op_cudnn.cu
  100. clip_op.cc
  101. clip_op.cu
  102. clip_op.h
  103. CMakeLists.txt
  104. collect_and_distribute_fpn_rpn_proposals_op.cc
  105. collect_and_distribute_fpn_rpn_proposals_op.h
  106. communicator_op.cc
  107. communicator_op_gpu.cc
  108. concat_split_op.cc
  109. concat_split_op.h
  110. concat_split_op_gpu.cc
  111. conditional_op.cc
  112. conditional_op.h
  113. conv_gradient_op.cc
  114. conv_op.cc
  115. conv_op.h
  116. conv_op_cache_cudnn.cc
  117. conv_op_cache_cudnn.h
  118. conv_op_cache_cudnn_test.cc
  119. conv_op_cudnn.cc
  120. conv_op_eigen.cc
  121. conv_op_gpu.cc
  122. conv_op_impl.h
  123. conv_op_shared.cc
  124. conv_op_shared.h
  125. conv_op_shared_gpu.cc
  126. conv_pool_op_base.h
  127. conv_transpose_gradient_op.cc
  128. conv_transpose_op.cc
  129. conv_transpose_op.h
  130. conv_transpose_op_cudnn.cc
  131. conv_transpose_op_gpu.cc
  132. conv_transpose_op_impl.h
  133. conv_transpose_op_mobile.cc
  134. conv_transpose_op_mobile.h
  135. conv_transpose_op_mobile_impl.h
  136. conv_transpose_op_mobile_test.cc
  137. conv_transpose_unpool_op_base.h
  138. copy_op.cc
  139. copy_op.cu
  140. copy_op.h
  141. copy_rows_to_tensor_op.cc
  142. copy_rows_to_tensor_op.h
  143. cos_op.cc
  144. cos_op.cu
  145. cos_op.h
  146. cosh_op.cc
  147. cosh_op.cu
  148. cosh_op.h
  149. cosine_embedding_criterion_op.cc
  150. cosine_embedding_criterion_op.cu
  151. cosine_embedding_criterion_op.h
  152. counter_ops.cc
  153. counter_ops.h
  154. counter_ops_gpu.cc
  155. crash_op.cc
  156. create_scope_op.cc
  157. create_scope_op.h
  158. crf_viterbi_op.cc
  159. cross_entropy_op.cc
  160. cross_entropy_op.cu
  161. cross_entropy_op.h
  162. ctc_beam_search_decoder_op.cc
  163. ctc_beam_search_decoder_op.h
  164. ctc_greedy_decoder_op.cc
  165. ctc_greedy_decoder_op.h
  166. cube_op.cc
  167. cube_op.cu
  168. cube_op.h
  169. data_couple.cc
  170. data_couple.h
  171. data_couple_gpu.cu
  172. dataset_ops.cc
  173. dataset_ops.h
  174. deform_conv_gradient_op.cc
  175. deform_conv_op.cc
  176. deform_conv_op.cu
  177. deform_conv_op.h
  178. deform_conv_op_impl.h
  179. dense_vector_to_id_list_op.cc
  180. dense_vector_to_id_list_op.h
  181. depthwise_3x3_conv_op_cudnn.cu
  182. distance_op.cc
  183. distance_op.cu
  184. distance_op.h
  185. do_op.cc
  186. do_op.h
  187. do_op_gpu.cc
  188. dropout_op.cc
  189. dropout_op.cu
  190. dropout_op.h
  191. dropout_op_cudnn.cc
  192. elementwise_add_gradient_op.cc
  193. elementwise_add_op.cc
  194. elementwise_add_op.h
  195. elementwise_add_op_gpu.cc
  196. elementwise_div_gradient_op.cc
  197. elementwise_div_op.cc
  198. elementwise_div_op.cu
  199. elementwise_div_op.h
  200. elementwise_linear_op.cc
  201. elementwise_linear_op.cu
  202. elementwise_linear_op.h
  203. elementwise_logical_ops.cc
  204. elementwise_logical_ops.h
  205. elementwise_mul_gradient_op.cc
  206. elementwise_mul_op.cc
  207. elementwise_mul_op.cu
  208. elementwise_mul_op.h
  209. elementwise_op_gpu_test.cc
  210. elementwise_op_test.cc
  211. elementwise_op_test.h
  212. elementwise_ops.cc
  213. elementwise_ops.cu
  214. elementwise_ops.h
  215. elementwise_ops_schema.cc
  216. elementwise_ops_utils.cc
  217. elementwise_ops_utils.h
  218. elementwise_sub_gradient_op.cc
  219. elementwise_sub_op.cc
  220. elementwise_sub_op.h
  221. elementwise_sub_op_gpu.cc
  222. elementwise_sum_op.cc
  223. elu_op.cc
  224. elu_op.cu
  225. elu_op.h
  226. elu_op_cudnn.cc
  227. enforce_finite_op.cc
  228. enforce_finite_op.cu
  229. enforce_finite_op.h
  230. ensure_clipped_op.cc
  231. ensure_clipped_op.h
  232. ensure_cpu_output_op.cc
  233. ensure_cpu_output_op.cu
  234. ensure_cpu_output_op.h
  235. erf_op.cc
  236. erf_op.cu
  237. erf_op.h
  238. exp_op.cc
  239. exp_op.h
  240. exp_op_gpu.cc
  241. expand_op.cc
  242. expand_op.h
  243. expand_op_gpu.cc
  244. expand_squeeze_dims_op.cc
  245. expand_squeeze_dims_op.h
  246. expand_squeeze_dims_op_gpu.cc
  247. fc_inference.cc
  248. fc_inference.h
  249. feature_maps_ops.cc
  250. feature_maps_ops.h
  251. feed_blob_op.cc
  252. feed_blob_op.h
  253. filler_op.cc
  254. filler_op.cu
  255. filler_op.h
  256. find_duplicate_elements_op.cc
  257. find_duplicate_elements_op.h
  258. find_op.cc
  259. find_op.cu
  260. find_op.h
  261. flatten_op.cc
  262. flatten_op.h
  263. flexible_top_k.cc
  264. flexible_top_k.h
  265. floor_op.cc
  266. floor_op.cu
  267. floor_op.h
  268. free_op.cc
  269. free_op.h
  270. free_op_gpu.cc
  271. fully_connected_op.cc
  272. fully_connected_op.h
  273. fully_connected_op_gpu.cc
  274. fused_rowwise_8bit_conversion_ops.cc
  275. fused_rowwise_8bit_conversion_ops.h
  276. fused_rowwise_nbit_conversion_ops.cc
  277. fused_rowwise_nbit_conversion_ops.h
  278. fused_rowwise_nbitfake_conversion_ops.cc
  279. fused_rowwise_nbitfake_conversion_ops.h
  280. fused_rowwise_random_quantization_ops.cc
  281. fused_rowwise_random_quantization_ops.h
  282. gather_fused_8bit_rowwise_op.cc
  283. gather_fused_8bit_rowwise_op.h
  284. gather_op.cc
  285. gather_op.cu
  286. gather_op.cuh
  287. gather_op.h
  288. gather_ranges_to_dense_op.cc
  289. gather_ranges_to_dense_op.h
  290. gelu_op.cc
  291. gelu_op.cu
  292. gelu_op.h
  293. generate_proposals_op.cc
  294. generate_proposals_op.cu
  295. generate_proposals_op.h
  296. generate_proposals_op_gpu_test.cc
  297. generate_proposals_op_test.cc
  298. generate_proposals_op_util_boxes.h
  299. generate_proposals_op_util_boxes_test.cc
  300. generate_proposals_op_util_nms.h
  301. generate_proposals_op_util_nms_gpu.cu
  302. generate_proposals_op_util_nms_gpu.h
  303. generate_proposals_op_util_nms_gpu_test.cc
  304. generate_proposals_op_util_nms_test.cc
  305. given_tensor_byte_string_to_uint8_fill_op.cc
  306. given_tensor_byte_string_to_uint8_fill_op.cu
  307. given_tensor_byte_string_to_uint8_fill_op.h
  308. given_tensor_fill_op.cc
  309. given_tensor_fill_op.cu
  310. given_tensor_fill_op.h
  311. glu_op.cc
  312. glu_op.cu
  313. glu_op.h
  314. group_norm_op.cc
  315. group_norm_op.cu
  316. group_norm_op.h
  317. gru_unit_op.cc
  318. gru_unit_op.h
  319. gru_unit_op_gpu.cu
  320. h_softmax_op.cc
  321. h_softmax_op.h
  322. half_float_ops.cc
  323. half_float_ops.cu
  324. half_float_ops.h
  325. half_float_ops_test.cc
  326. hard_sigmoid_op.cc
  327. hard_sigmoid_op.cu
  328. hard_sigmoid_op.h
  329. heatmap_max_keypoint_op.cc
  330. heatmap_max_keypoint_op.h
  331. if_op.cc
  332. if_op.h
  333. if_op_gpu.cc
  334. im2col_op.cc
  335. im2col_op.h
  336. im2col_op_gpu.cc
  337. index_hash_ops.cc
  338. index_hash_ops.h
  339. index_ops.cc
  340. index_ops.h
  341. inference_lstm_op.cc
  342. inference_lstm_op.h
  343. instance_norm_gradient_op.cc
  344. instance_norm_op.cc
  345. instance_norm_op.cu
  346. instance_norm_op.h
  347. integral_image_op.cc
  348. integral_image_op.cu
  349. integral_image_op.h
  350. is_empty_op.cc
  351. is_empty_op.h
  352. jsd_op.cc
  353. jsd_op.h
  354. key_split_ops.cc
  355. key_split_ops.h
  356. last_n_window_collector.cc
  357. layer_norm_op.cc
  358. layer_norm_op.cu
  359. layer_norm_op.h
  360. leaky_relu_op.cc
  361. leaky_relu_op.cu
  362. leaky_relu_op.h
  363. length_split_op.cc
  364. length_split_op.h
  365. lengths_pad_op.cc
  366. lengths_pad_op.cu
  367. lengths_pad_op.h
  368. lengths_reducer_fused_8bit_rowwise_ops.cc
  369. lengths_reducer_fused_8bit_rowwise_ops.h
  370. lengths_reducer_fused_nbit_rowwise_ops.cc
  371. lengths_reducer_fused_nbit_rowwise_ops.h
  372. lengths_reducer_ops.cc
  373. lengths_reducer_ops.h
  374. lengths_reducer_rowwise_8bit_ops.cc
  375. lengths_reducer_rowwise_8bit_ops.h
  376. lengths_tile_op.cc
  377. lengths_tile_op.cu
  378. lengths_tile_op.h
  379. lengths_top_k_op.cc
  380. lengths_top_k_op.h
  381. listwise_l2r_op.cc
  382. listwise_l2r_op.h
  383. load_save_op.cc
  384. load_save_op.h
  385. load_save_op_gpu.cc
  386. load_save_op_util.cc
  387. load_save_op_util.h
  388. local_response_normalization_op.cc
  389. local_response_normalization_op.cu
  390. local_response_normalization_op.h
  391. local_response_normalization_op_cudnn.cc
  392. locally_connected_op.cc
  393. locally_connected_op.h
  394. locally_connected_op_gpu.cc
  395. locally_connected_op_impl.h
  396. locally_connected_op_util.cc
  397. locally_connected_op_util.h
  398. log_op.cc
  399. log_op.h
  400. log_op_gpu.cc
  401. logit_op.cc
  402. logit_op.cu
  403. logit_op.h
  404. loss_op.cc
  405. loss_op.cu
  406. loss_op.h
  407. lp_pool_op.cc
  408. lp_pool_op.cu
  409. lpnorm_op.cc
  410. lpnorm_op.h
  411. lstm_unit_op.cc
  412. lstm_unit_op.h
  413. lstm_unit_op_gpu.cu
  414. lstm_utils.h
  415. map_ops.cc
  416. map_ops.h
  417. margin_ranking_criterion_op.cc
  418. margin_ranking_criterion_op.cu
  419. margin_ranking_criterion_op.h
  420. matmul_op.cc
  421. matmul_op.h
  422. matmul_op_gpu.cc
  423. max_pool_with_index.cu
  424. max_pool_with_index_gpu.h
  425. mean_op.cc
  426. mean_op.cu
  427. mean_op.h
  428. mem_query_op.cu
  429. merge_id_lists_op.cc
  430. merge_id_lists_op.h
  431. minmax_gradient_ops.cc
  432. minmax_ops.cc
  433. minmax_ops.cu
  434. minmax_ops.h
  435. mod_op.cc
  436. mod_op.h
  437. moments_op.cc
  438. moments_op.cu
  439. moments_op.h
  440. multi_class_accuracy_op.cc
  441. multi_class_accuracy_op.cu
  442. multi_class_accuracy_op.h
  443. negate_gradient_op.cc
  444. negate_gradient_op.h
  445. negate_gradient_op_gpu.cc
  446. negative_op.cc
  447. negative_op.h
  448. negative_op_gpu.cc
  449. ngram_ops.cc
  450. ngram_ops.h
  451. no_default_engine_op.h
  452. norm_planar_yuv_op.cc
  453. normalize_l1_op.cc
  454. normalize_l1_op.h
  455. normalize_op.cc
  456. normalize_op.h
  457. normalize_ops.cu
  458. numpy_tile_op.cc
  459. numpy_tile_op.h
  460. one_hot_ops.cc
  461. one_hot_ops.cu
  462. one_hot_ops.h
  463. onnx_while_op.cc
  464. onnx_while_op.h
  465. op_utils_cudnn.h
  466. operator_fallback_gpu.h
  467. operator_fallback_gpu_test.cc
  468. order_switch_ops.cc
  469. order_switch_ops.h
  470. order_switch_ops_cudnn.cc
  471. order_switch_ops_gpu.cc
  472. pack_rnn_sequence_op.cc
  473. pack_rnn_sequence_op.h
  474. pack_segments.cc
  475. pack_segments.cu
  476. pack_segments.h
  477. pad_op.cc
  478. pad_op.h
  479. pad_op_gpu.cu
  480. partition_ops.cc
  481. partition_ops.h
  482. percentile_op.cc
  483. percentile_op.h
  484. perplexity_op.cc
  485. perplexity_op.cu
  486. perplexity_op.h
  487. piecewise_linear_transform_op.cc
  488. piecewise_linear_transform_op.cu
  489. piecewise_linear_transform_op.h
  490. pool_gradient_op.cc
  491. pool_op.cc
  492. pool_op.cu
  493. pool_op.h
  494. pool_op_cudnn.cc
  495. pool_op_util.cc
  496. pool_op_util.h
  497. pow_op.cc
  498. pow_op.cu
  499. pow_op.h
  500. prefetch_op.h
  501. prelu_op.cc
  502. prelu_op.cu
  503. prelu_op.h
  504. prepend_dim_op.cc
  505. prepend_dim_op.h
  506. prepend_dim_op_gpu.cc
  507. quant_decode_op.cc
  508. quant_decode_op.h
  509. quantile_op.cc
  510. quantile_op.h
  511. rank_loss_op.cc
  512. rank_loss_op.h
  513. reciprocal_gradient_op.cc
  514. reciprocal_op.cc
  515. reciprocal_op.cu
  516. reciprocal_op.h
  517. reduce_front_back_max_ops.cc
  518. reduce_front_back_max_ops.cu
  519. reduce_front_back_max_ops.h
  520. reduce_front_back_mean_ops.cc
  521. reduce_front_back_sum_mean_ops.cu
  522. reduce_front_back_sum_mean_ops.h
  523. reduce_front_back_sum_ops.cc
  524. reduce_ops.cc
  525. reduce_ops.cu
  526. reduce_ops.h
  527. reducer_functors.h
  528. reduction_ops.cc
  529. reduction_ops.cu
  530. reduction_ops.h
  531. relu_n_op.cc
  532. relu_n_op.cu
  533. relu_n_op.h
  534. relu_op.cc
  535. relu_op.cu
  536. relu_op.h
  537. remove_data_blocks_op.cc
  538. remove_data_blocks_op.h
  539. replace_nan_op.cc
  540. replace_nan_op.cu
  541. replace_nan_op.h
  542. reservoir_sampling.cc
  543. reshape_op.cc
  544. reshape_op.h
  545. reshape_op_gpu.cc
  546. reshape_op_gpu_test.cc
  547. resize_3d_op.cc
  548. resize_3d_op.cu
  549. resize_3d_op.h
  550. resize_op.cc
  551. resize_op.cu
  552. resize_op.h
  553. reverse_packed_segs_op.cc
  554. reverse_packed_segs_op.cu
  555. reverse_packed_segs_op.h
  556. rmac_regions_op.cc
  557. rmac_regions_op.cu
  558. rmac_regions_op.h
  559. roi_align_gradient_op.cc
  560. roi_align_gradient_op.cu
  561. roi_align_gradient_op.h
  562. roi_align_op.cc
  563. roi_align_op.cu
  564. roi_align_op.h
  565. roi_align_op_gpu_test.cc
  566. roi_align_rotated_gradient_op.cc
  567. roi_align_rotated_gradient_op.cu
  568. roi_align_rotated_gradient_op.h
  569. roi_align_rotated_op.cc
  570. roi_align_rotated_op.cu
  571. roi_align_rotated_op.h
  572. roi_pool_op.cc
  573. roi_pool_op.cu
  574. roi_pool_op.h
  575. rowmul_op.cc
  576. rowmul_op.h
  577. rsqrt_op.cc
  578. rsqrt_op.cu
  579. rsqrt_op.h
  580. scale_blobs_op.cc
  581. scale_blobs_op.cu
  582. scale_blobs_op.h
  583. scale_op.cc
  584. scale_op.h
  585. scale_op_gpu.cc
  586. segment_reduction_op.cc
  587. segment_reduction_op.h
  588. segment_reduction_op_gpu.cu
  589. selu_op.cc
  590. selu_op.cu
  591. selu_op.h
  592. sequence_ops.cc
  593. sequence_ops.cu
  594. sequence_ops.h
  595. shape_op.cc
  596. shape_op.h
  597. shape_op_gpu.cc
  598. sigmoid_gradient_op.cc
  599. sigmoid_op.cc
  600. sigmoid_op.cu
  601. sigmoid_op.h
  602. sigmoid_op_cudnn.cc
  603. sin_op.cc
  604. sin_op.cu
  605. sin_op.h
  606. sinh_op.cc
  607. sinh_op.cu
  608. sinh_op.h
  609. sinusoid_position_encoding_op.cc
  610. sinusoid_position_encoding_op.h
  611. slice_op.cc
  612. slice_op.cu
  613. slice_op.h
  614. softmax_op.cc
  615. softmax_op.h
  616. softmax_op_cudnn.cc
  617. softmax_ops.cu
  618. softmax_utils.cc
  619. softmax_utils.h
  620. softmax_with_loss_op.cc
  621. softmax_with_loss_op.h
  622. softplus_op.cc
  623. softplus_op.cu
  624. softplus_op.h
  625. softsign_op.cc
  626. softsign_op.cu
  627. softsign_op.h
  628. space_batch_op.cc
  629. space_batch_op.h
  630. space_batch_op_gpu.cu
  631. sparse_dropout_with_replacement_op.cc
  632. sparse_dropout_with_replacement_op.h
  633. sparse_normalize_op.cc
  634. sparse_normalize_op.h
  635. sparse_normalize_op_gpu.cu
  636. sparse_to_dense_mask_op.cc
  637. sparse_to_dense_mask_op.h
  638. sparse_to_dense_op.cc
  639. sparse_to_dense_op.cu
  640. sparse_to_dense_op.h
  641. spatial_batch_norm_gradient_op.cc
  642. spatial_batch_norm_op.cc
  643. spatial_batch_norm_op.cu
  644. spatial_batch_norm_op.h
  645. spatial_batch_norm_op_cudnn.cu
  646. spatial_batch_norm_op_impl.cuh
  647. spatial_softmax_with_loss_op.cc
  648. spatial_softmax_with_loss_op.h
  649. sqr_op.cc
  650. sqr_op.h
  651. sqr_op_gpu.cc
  652. sqrt_op.cc
  653. sqrt_op.h
  654. sqrt_op_gpu.cc
  655. square_root_divide_op.cc
  656. square_root_divide_op.h
  657. stats_ops.cc
  658. stats_put_ops.cc
  659. stats_put_ops.h
  660. stop_gradient.cc
  661. stop_gradient.h
  662. stop_gradient_gpu.cc
  663. string_ops.cc
  664. string_ops.h
  665. string_ops_test.cc
  666. stump_func_op.cc
  667. stump_func_op.cu
  668. stump_func_op.h
  669. stylizer_ops.cc
  670. summarize_op.cc
  671. summarize_op.cu
  672. summarize_op.h
  673. swish_op.cc
  674. swish_op.cu
  675. swish_op.h
  676. tan_op.cc
  677. tan_op.cu
  678. tan_op.h
  679. tanh_gradient_op.cc
  680. tanh_op.cc
  681. tanh_op.cu
  682. tanh_op.h
  683. tanh_op_cudnn.cc
  684. tensor_protos_db_input.cc
  685. tensor_protos_db_input.h
  686. tensor_protos_db_input_gpu.cc
  687. text_file_reader.cc
  688. text_file_reader_utils.cc
  689. text_file_reader_utils.h
  690. text_file_reader_utils_test.cc
  691. thresholded_relu_op.cc
  692. thresholded_relu_op.cu
  693. thresholded_relu_op.h
  694. tile_op.cc
  695. tile_op.cu
  696. tile_op.h
  697. top_k.cc
  698. top_k.cu
  699. top_k.h
  700. top_k_heap_selection.cuh
  701. top_k_radix_selection.cuh
  702. transpose_op.cc
  703. transpose_op.cu
  704. transpose_op.h
  705. transpose_op_cudnn.cc
  706. tt_linear_op.cc
  707. tt_linear_op.h
  708. unique_ops.cc
  709. unique_ops.cu
  710. unique_ops.h
  711. upsample_op.cc
  712. upsample_op.cu
  713. upsample_op.h
  714. utility_ops.cc
  715. utility_ops.cu
  716. utility_ops.h
  717. utility_ops_gpu_test.cc
  718. utility_ops_test.cc
  719. variable_length_sequence_padding.cc
  720. variable_length_sequence_padding.h
  721. weighted_multi_sampling_op.cc
  722. weighted_multi_sampling_op.h
  723. weighted_sample_op.cc
  724. weighted_sample_op.cu
  725. weighted_sample_op.h
  726. while_op.cc
  727. while_op.h
  728. while_op_gpu.cc
  729. workspace_ops.cc
  730. zero_gradient_op.cc
  731. zero_gradient_op.h
  732. zero_gradient_op_gpu.cc