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