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