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