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