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