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