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