Google Git
Sign in
android / platform / external / pytorch / d38adfe35d / . / caffe2 / operators
tree: 8d34909cb0fe9a451b91dee9a814933978c10adb [path history] [tgz]
  1. rnn/
  2. abs_op.cc
  3. abs_op.cu
  4. accumulate_op.cc
  5. accumulate_op.cu
  6. accumulate_op.h
  7. accuracy_op.cc
  8. accuracy_op.cu
  9. accuracy_op.h
  10. apmeter_op.cc
  11. apmeter_op.h
  12. arg_max_op.cc
  13. arg_max_op.h
  14. arg_ops.cc
  15. arg_ops.cu
  16. arg_ops.h
  17. arg_ops_eigen.h
  18. assert_op.cc
  19. assert_op.cu
  20. assert_op.h
  21. atomic_ops.cc
  22. batch_box_cox_op.cc
  23. batch_box_cox_op.h
  24. batch_gather_ops.cc
  25. batch_gather_ops.cu
  26. batch_gather_ops.h
  27. batch_matmul_op.cc
  28. batch_matmul_op.cu
  29. batch_matmul_op.h
  30. batch_matmul_op_gpu_test.cc
  31. batch_matmul_op_test.cc
  32. batch_sparse_to_dense_op.cc
  33. batch_sparse_to_dense_op.h
  34. bbox_transform_op.cc
  35. bbox_transform_op.h
  36. boolean_mask_ops.cc
  37. boolean_mask_ops.cu
  38. boolean_mask_ops.h
  39. boolean_unmask_ops.cc
  40. boolean_unmask_ops.cu
  41. boolean_unmask_ops.h
  42. boolean_unmask_ops_test.cc
  43. box_with_nms_limit_op.cc
  44. box_with_nms_limit_op.h
  45. cast_op.cc
  46. cast_op.cu
  47. cast_op.h
  48. ceil_op.cc
  49. ceil_op.cu
  50. ceil_op.h
  51. channel_backprop_stats_op.cc
  52. channel_backprop_stats_op.cu
  53. channel_backprop_stats_op.h
  54. channel_shuffle_op.cc
  55. channel_shuffle_op.h
  56. channel_shuffle_op_gpu.cu
  57. channel_stats_op.cc
  58. channel_stats_op.cu
  59. channel_stats_op.h
  60. clip_op.cc
  61. clip_op.cu
  62. clip_op.h
  63. CMakeLists.txt
  64. collect_and_distribute_fpn_rpn_proposals_op.cc
  65. collect_and_distribute_fpn_rpn_proposals_op.h
  66. communicator_op.cc
  67. communicator_op_gpu.cc
  68. concat_split_op.cc
  69. concat_split_op.h
  70. concat_split_op_gpu.cc
  71. conditional_op.cc
  72. conditional_op.h
  73. conv_gradient_op.cc
  74. conv_op.cc
  75. conv_op.h
  76. conv_op_cache_cudnn.cc
  77. conv_op_cache_cudnn.h
  78. conv_op_cache_cudnn_test.cc
  79. conv_op_cudnn.cc
  80. conv_op_eigen.cc
  81. conv_op_gpu.cc
  82. conv_op_impl.h
  83. conv_op_shared.cc
  84. conv_op_shared.h
  85. conv_op_shared_gpu.cc
  86. conv_pool_op_base.h
  87. conv_transpose_gradient_op.cc
  88. conv_transpose_op.cc
  89. conv_transpose_op.h
  90. conv_transpose_op_cudnn.cc
  91. conv_transpose_op_gpu.cc
  92. conv_transpose_op_impl.h
  93. conv_transpose_op_mobile.cc
  94. conv_transpose_op_mobile.h
  95. conv_transpose_op_mobile_impl.h
  96. conv_transpose_op_mobile_test.cc
  97. conv_transpose_unpool_op_base.h
  98. cos_op.cc
  99. cos_op.cu
  100. cosine_embedding_criterion_op.cc
  101. cosine_embedding_criterion_op.cu
  102. cosine_embedding_criterion_op.h
  103. counter_ops.cc
  104. counter_ops.h
  105. counter_ops_gpu.cc
  106. create_scope_op.cc
  107. create_scope_op.h
  108. cross_entropy_op.cc
  109. cross_entropy_op.cu
  110. cross_entropy_op.h
  111. dataset_ops.cc
  112. dataset_ops.h
  113. deform_conv_gradient_op.cc
  114. deform_conv_op.cc
  115. deform_conv_op.cu
  116. deform_conv_op.h
  117. deform_conv_op_impl.h
  118. distance_op.cc
  119. distance_op.cu
  120. distance_op.h
  121. do_op.cc
  122. do_op.h
  123. do_op_gpu.cc
  124. dropout_op.cc
  125. dropout_op.cu
  126. dropout_op.h
  127. dropout_op_cudnn.cc
  128. elementwise_add_op.cc
  129. elementwise_div_op.cc
  130. elementwise_linear_op.cc
  131. elementwise_linear_op.cu
  132. elementwise_linear_op.h
  133. elementwise_logical_ops.cc
  134. elementwise_logical_ops.h
  135. elementwise_mul_op.cc
  136. elementwise_op.cc
  137. elementwise_op.cu
  138. elementwise_op.h
  139. elementwise_op_gpu_test.cc
  140. elementwise_op_schema.cc
  141. elementwise_op_test.cc
  142. elementwise_op_test.h
  143. elementwise_sub_op.cc
  144. elementwise_sum_op.cc
  145. elu_op.cc
  146. elu_op.cu
  147. elu_op.h
  148. exp_op.cc
  149. exp_op.cu
  150. expand_squeeze_dims_op.cc
  151. expand_squeeze_dims_op.h
  152. expand_squeeze_dims_op_gpu.cc
  153. extend_tensor_op.cc
  154. fc_inference.cc
  155. fc_inference.h
  156. feed_blob_op.cc
  157. feed_blob_op.h
  158. filler_op.cc
  159. filler_op.cu
  160. filler_op.h
  161. filler_op_gpu.cc
  162. find_duplicate_elements_op.cc
  163. find_duplicate_elements_op.h
  164. find_op.cc
  165. find_op.cu
  166. find_op.h
  167. flatten_op.cc
  168. flatten_op.h
  169. flexible_top_k.cc
  170. flexible_top_k.h
  171. floor_op.cc
  172. floor_op.cu
  173. floor_op.h
  174. free_op.cc
  175. free_op.h
  176. free_op_gpu.cc
  177. fully_connected_op.cc
  178. fully_connected_op.h
  179. fully_connected_op_gpu.cc
  180. fused_rowwise_8bit_conversion_ops.cc
  181. fused_rowwise_8bit_conversion_ops.h
  182. gather_fused_8bit_rowwise_op.cc
  183. gather_fused_8bit_rowwise_op.h
  184. gather_ranges_to_dense_op.cc
  185. gather_ranges_to_dense_op.h
  186. generate_proposals_op.cc
  187. generate_proposals_op.h
  188. generate_proposals_op_test.cc
  189. generate_proposals_op_util_boxes.h
  190. generate_proposals_op_util_boxes_test.cc
  191. generate_proposals_op_util_nms.h
  192. generate_proposals_op_util_nms_test.cc
  193. given_tensor_fill_op.cc
  194. given_tensor_fill_op.cu
  195. given_tensor_fill_op.h
  196. glu_op.cc
  197. glu_op.cu
  198. glu_op.h
  199. gru_unit_op.cc
  200. gru_unit_op.h
  201. gru_unit_op_gpu.cu
  202. h_softmax_op.cc
  203. h_softmax_op.h
  204. half_float_ops.cc
  205. half_float_ops.cu
  206. half_float_ops.h
  207. if_op.cc
  208. if_op.h
  209. if_op_gpu.cc
  210. im2col_op.cc
  211. im2col_op.h
  212. im2col_op_gpu.cc
  213. index_hash_ops.cc
  214. index_hash_ops.h
  215. index_ops.cc
  216. instance_norm_gradient_op.cc
  217. instance_norm_op.cc
  218. instance_norm_op.cu
  219. instance_norm_op.h
  220. jsd_op.cc
  221. jsd_op.h
  222. key_split_ops.cc
  223. key_split_ops.h
  224. last_n_window_collector.cc
  225. layer_norm_op.cc
  226. layer_norm_op.cu
  227. layer_norm_op.h
  228. leaky_relu_op.cc
  229. leaky_relu_op.cu
  230. leaky_relu_op.h
  231. lengths_reducer_fused_8bit_rowwise_ops.cc
  232. lengths_reducer_fused_8bit_rowwise_ops.h
  233. lengths_reducer_ops.cc
  234. lengths_reducer_ops.h
  235. lengths_reducer_rowwise_8bit_ops.cc
  236. lengths_reducer_rowwise_8bit_ops.h
  237. lengths_tile_op.cc
  238. lengths_tile_op.h
  239. lengths_tile_op_gpu.cc
  240. lengths_top_k_op.cc
  241. lengths_top_k_op.h
  242. listwise_l2r_op.cc
  243. listwise_l2r_op.h
  244. load_save_op.cc
  245. load_save_op.h
  246. load_save_op_gpu.cc
  247. local_response_normalization_op.cc
  248. local_response_normalization_op.cu
  249. local_response_normalization_op.h
  250. local_response_normalization_op_cudnn.cc
  251. locally_connected_op.cc
  252. locally_connected_op.h
  253. locally_connected_op_gpu.cc
  254. locally_connected_op_impl.h
  255. locally_connected_op_util.cc
  256. locally_connected_op_util.h
  257. log_op.cc
  258. log_op.cu
  259. logit_op.cc
  260. logit_op.cu
  261. logit_op.h
  262. loss_op.cc
  263. loss_op.cu
  264. loss_op.h
  265. lp_pool_op.cc
  266. lp_pool_op.cu
  267. lpnorm_op.cc
  268. lpnorm_op.h
  269. lstm_unit_op.cc
  270. lstm_unit_op.h
  271. lstm_unit_op_gpu.cu
  272. map_ops.cc
  273. map_ops.h
  274. margin_ranking_criterion_op.cc
  275. margin_ranking_criterion_op.cu
  276. margin_ranking_criterion_op.h
  277. math_ops.cc
  278. math_ops.cu
  279. math_ops.h
  280. matmul_op.cc
  281. matmul_op.h
  282. matmul_op_gpu.cc
  283. max_pool_with_index.cu
  284. max_pool_with_index.h
  285. mean_op.cc
  286. mean_op.cu
  287. mean_op.h
  288. mem_query_op.cu
  289. merge_id_lists_op.cc
  290. merge_id_lists_op.h
  291. minmax_gradient_ops.cc
  292. minmax_ops.cc
  293. minmax_ops.h
  294. mod_op.cc
  295. mod_op.h
  296. multi_class_accuracy_op.cc
  297. multi_class_accuracy_op.cu
  298. multi_class_accuracy_op.h
  299. negate_gradient_op.cc
  300. negate_gradient_op.h
  301. negate_gradient_op_gpu.cc
  302. negative_op.cc
  303. negative_op.cu
  304. ngram_ops.cc
  305. ngram_ops.h
  306. no_default_engine_op.h
  307. norm_planar_yuv_op.cc
  308. normalize_l1_op.cc
  309. normalize_l1_op.h
  310. normalize_op.cc
  311. normalize_op.h
  312. normalize_ops.cu
  313. one_hot_ops.cc
  314. one_hot_ops.cu
  315. one_hot_ops.h
  316. onnx_while_op.cc
  317. onnx_while_op.h
  318. op_utils_cudnn.h
  319. operator_fallback_gpu.h
  320. operator_fallback_gpu_test.cc
  321. order_switch_ops.cc
  322. order_switch_ops.cu
  323. order_switch_ops.h
  324. pack_rnn_sequence_op.cc
  325. pack_rnn_sequence_op.h
  326. pack_segments.cc
  327. pack_segments.cu
  328. pack_segments.h
  329. pad_op.cc
  330. pad_op.h
  331. pad_op_gpu.cu
  332. partition_ops.cc
  333. partition_ops.h
  334. percentile_op.cc
  335. percentile_op.h
  336. perplexity_op.cc
  337. perplexity_op.cu
  338. perplexity_op.h
  339. piecewise_linear_transform_op.cc
  340. piecewise_linear_transform_op.cu
  341. piecewise_linear_transform_op.h
  342. pool_gradient_op.cc
  343. pool_op.cc
  344. pool_op.cu
  345. pool_op.h
  346. pool_op_cudnn.cu
  347. pow_op.cc
  348. pow_op.cu
  349. pow_op.h
  350. prefetch_op.h
  351. prelu_op.cc
  352. prelu_op.cu
  353. prelu_op.h
  354. prepend_dim_op.cc
  355. prepend_dim_op.h
  356. prepend_dim_op_gpu.cc
  357. quant_decode_op.cc
  358. quant_decode_op.h
  359. rank_loss_op.cc
  360. rank_loss_op.h
  361. reduce_ops.cc
  362. reduce_ops.h
  363. reducer_functors.h
  364. reduction_front_back_ops.cc
  365. reduction_front_back_ops.cu
  366. reduction_front_back_ops.h
  367. reduction_ops.cc
  368. reduction_ops.cu
  369. reduction_ops.h
  370. relu_op.cc
  371. relu_op.cu
  372. relu_op.h
  373. relu_op_cudnn.cc
  374. relu_op_fp16.cu
  375. remove_data_blocks_op.cc
  376. remove_data_blocks_op.h
  377. replace_nan_op.cc
  378. replace_nan_op.cu
  379. replace_nan_op.h
  380. reservoir_sampling.cc
  381. reshape_op.cc
  382. reshape_op.h
  383. reshape_op_gpu.cc
  384. reshape_op_gpu_test.cc
  385. resize_op.cc
  386. resize_op.cu
  387. resize_op.h
  388. reverse_packed_segs_op.cc
  389. reverse_packed_segs_op.cu
  390. reverse_packed_segs_op.h
  391. rmac_regions_op.cc
  392. rmac_regions_op.cu
  393. rmac_regions_op.h
  394. roi_align_gradient_op.cc
  395. roi_align_gradient_op.cu
  396. roi_align_gradient_op.h
  397. roi_align_op.cc
  398. roi_align_op.cu
  399. roi_align_op.h
  400. roi_align_op_gpu_test.cc
  401. roi_pool_op.cc
  402. roi_pool_op.cu
  403. roi_pool_op.h
  404. rowmul_op.cc
  405. rowmul_op.h
  406. scale_op.cc
  407. scale_op.h
  408. scale_op_gpu.cc
  409. segment_reduction_op.cc
  410. segment_reduction_op.h
  411. segment_reduction_op_gpu.cu
  412. selu_op.cc
  413. selu_op.cu
  414. selu_op.h
  415. sequence_ops.cc
  416. sequence_ops.cu
  417. sequence_ops.h
  418. shape_op.cc
  419. shape_op.h
  420. shape_op_gpu.cc
  421. sigmoid_op.cc
  422. sigmoid_op.cu
  423. sin_op.cc
  424. sin_op.cu
  425. sinusoid_position_encoding_op.cc
  426. sinusoid_position_encoding_op.h
  427. slice_op.cc
  428. slice_op.cu
  429. slice_op.h
  430. softmax_op.cc
  431. softmax_op.h
  432. softmax_op_cudnn.cc
  433. softmax_ops.cu
  434. softmax_shared.cc
  435. softmax_shared.h
  436. softmax_with_loss_op.cc
  437. softmax_with_loss_op.h
  438. softplus_op.cc
  439. softplus_op.cu
  440. softplus_op.h
  441. softsign_op.cc
  442. softsign_op.cu
  443. space_batch_op.cc
  444. space_batch_op.h
  445. space_batch_op_gpu.cu
  446. sparse_normalize_op.cc
  447. sparse_normalize_op.h
  448. sparse_to_dense_mask_op.cc
  449. sparse_to_dense_mask_op.h
  450. sparse_to_dense_op.cc
  451. sparse_to_dense_op.cu
  452. sparse_to_dense_op.h
  453. spatial_batch_norm_gradient_op.cc
  454. spatial_batch_norm_op.cc
  455. spatial_batch_norm_op.h
  456. spatial_batch_norm_op_cudnn.cc
  457. spatial_softmax_with_loss_op.cc
  458. spatial_softmax_with_loss_op.h
  459. sqrt_op.cc
  460. sqrt_op.cu
  461. square_root_divide_op.cc
  462. square_root_divide_op.h
  463. stats_ops.cc
  464. stop_gradient.cc
  465. stop_gradient.h
  466. stop_gradient_gpu.cc
  467. string_ops.cc
  468. string_ops.h
  469. string_ops_test.cc
  470. stump_func_op.cc
  471. stump_func_op.cu
  472. stump_func_op.h
  473. stylizer_ops.cc
  474. summarize_op.cc
  475. summarize_op.cu
  476. summarize_op.h
  477. swish_op.cc
  478. swish_op.cu
  479. swish_op.h
  480. tanh_op.cc
  481. tanh_op.cu
  482. tensor_protos_db_input.cc
  483. tensor_protos_db_input.h
  484. tensor_protos_db_input_gpu.cc
  485. text_file_reader.cc
  486. text_file_reader_utils.cc
  487. text_file_reader_utils.h
  488. text_file_reader_utils_test.cc
  489. thresholded_relu_op.cc
  490. thresholded_relu_op.cu
  491. thresholded_relu_op.h
  492. tile_op.cc
  493. tile_op.cu
  494. tile_op.h
  495. top_k.cc
  496. top_k.cu
  497. top_k.h
  498. top_k_heap_selection.cuh
  499. top_k_radix_selection.cuh
  500. transpose_op.cc
  501. transpose_op.cu
  502. transpose_op.h
  503. transpose_op_cudnn.cc
  504. tt_linear_op.cc
  505. tt_linear_op.h
  506. utility_ops.cc
  507. utility_ops.cu
  508. utility_ops.h
  509. utility_ops_gpu.cc
  510. utility_ops_gpu_test.cc
  511. utility_ops_test.cc
  512. variable_length_sequence_padding.cc
  513. variable_length_sequence_padding.h
  514. weighted_multi_sampling_op.cc
  515. weighted_multi_sampling_op.h
  516. weighted_sample_op.cc
  517. weighted_sample_op.cu
  518. weighted_sample_op.h
  519. while_op.cc
  520. while_op.h
  521. while_op_gpu.cc
  522. workspace_ops.cc
  523. zero_gradient_op.cc
  524. zero_gradient_op.h
  525. zero_gradient_op_gpu.cc