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