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