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