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