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