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