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