Google Git
Sign in
android / platform / external / pytorch / 43adc5ba05 / . / caffe2 / operators
tree: 3475cc4a750d11a067eed89c4cce1a94d7782719 [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. atomic_ops.cc
  12. batch_box_cox_op.cc
  13. batch_box_cox_op.h
  14. batch_gather_ops.cc
  15. batch_gather_ops.cu
  16. batch_gather_ops.h
  17. batch_matmul_op.cc
  18. batch_matmul_op.cu
  19. batch_matmul_op.h
  20. boolean_mask_ops.cc
  21. boolean_mask_ops.cu
  22. boolean_mask_ops.h
  23. boolean_unmask_ops.cc
  24. boolean_unmask_ops.cu
  25. boolean_unmask_ops.h
  26. boolean_unmask_ops_test.cc
  27. cast_op.cc
  28. cast_op.cu
  29. cast_op.h
  30. channel_shuffle_op.cc
  31. channel_shuffle_op.h
  32. channel_shuffle_op_gpu.cu
  33. clip_op.cc
  34. clip_op.cu
  35. clip_op.h
  36. CMakeLists.txt
  37. communicator_op.cc
  38. communicator_op_gpu.cc
  39. concat_split_op.cc
  40. concat_split_op.h
  41. concat_split_op_gpu.cc
  42. conditional_op.cc
  43. conditional_op.h
  44. conv_gradient_op.cc
  45. conv_op.cc
  46. conv_op.h
  47. conv_op_cache_cudnn.cc
  48. conv_op_cache_cudnn.h
  49. conv_op_cache_cudnn_test.cc
  50. conv_op_cudnn.cc
  51. conv_op_eigen.cc
  52. conv_op_gpu.cc
  53. conv_op_impl.h
  54. conv_op_shared.cc
  55. conv_op_shared.h
  56. conv_op_shared_gpu.cc
  57. conv_pool_op_base.h
  58. conv_transpose_gradient_op.cc
  59. conv_transpose_op.cc
  60. conv_transpose_op.h
  61. conv_transpose_op_cudnn.cc
  62. conv_transpose_op_gpu.cc
  63. conv_transpose_op_impl.h
  64. conv_transpose_op_mobile.cc
  65. conv_transpose_op_mobile.h
  66. conv_transpose_op_mobile_impl.h
  67. conv_transpose_op_mobile_test.cc
  68. conv_transpose_unpool_op_base.h
  69. cos_op.cc
  70. cos_op.cu
  71. cosine_embedding_criterion_op.cc
  72. cosine_embedding_criterion_op.cu
  73. cosine_embedding_criterion_op.h
  74. counter_ops.cc
  75. counter_ops.h
  76. counter_ops_gpu.cc
  77. create_scope_op.cc
  78. create_scope_op.h
  79. cross_entropy_op.cc
  80. cross_entropy_op.cu
  81. cross_entropy_op.h
  82. dataset_ops.cc
  83. dataset_ops.h
  84. deform_conv_gradient_op.cc
  85. deform_conv_op.cc
  86. deform_conv_op.cu
  87. deform_conv_op.h
  88. deform_conv_op_impl.h
  89. distance_op.cc
  90. distance_op.cu
  91. distance_op.h
  92. do_op.cc
  93. do_op.h
  94. dropout_op.cc
  95. dropout_op.cu
  96. dropout_op.h
  97. dropout_op_cudnn.cc
  98. elementwise_add_op.cc
  99. elementwise_div_op.cc
  100. elementwise_linear_op.cc
  101. elementwise_linear_op.cu
  102. elementwise_linear_op.h
  103. elementwise_logical_ops.cc
  104. elementwise_logical_ops.h
  105. elementwise_mul_op.cc
  106. elementwise_op.cc
  107. elementwise_op.cu
  108. elementwise_op.h
  109. elementwise_op_gpu_test.cc
  110. elementwise_op_schema.cc
  111. elementwise_op_test.cc
  112. elementwise_op_test.h
  113. elementwise_sub_op.cc
  114. elementwise_sum_op.cc
  115. elu_op.cc
  116. elu_op.cu
  117. elu_op.h
  118. exp_op.cc
  119. exp_op.cu
  120. expand_squeeze_dims_op.cc
  121. expand_squeeze_dims_op.h
  122. expand_squeeze_dims_op_gpu.cc
  123. extend_tensor_op.cc
  124. feed_blob_op.cc
  125. feed_blob_op.h
  126. filler_op.cc
  127. filler_op.cu
  128. filler_op.h
  129. find_duplicate_elements_op.cc
  130. find_duplicate_elements_op.h
  131. find_op.cc
  132. find_op.cu
  133. find_op.h
  134. free_op.cc
  135. free_op.h
  136. free_op_gpu.cc
  137. fully_connected_op.cc
  138. fully_connected_op.h
  139. fully_connected_op_gpu.cc
  140. fully_connected_op_gpu_test.cc
  141. fully_connected_op_test.cc
  142. gather_ranges_to_dense_op.cc
  143. gather_ranges_to_dense_op.h
  144. given_tensor_fill_op.cc
  145. given_tensor_fill_op.cu
  146. given_tensor_fill_op.h
  147. gru_unit_op.cc
  148. gru_unit_op.h
  149. gru_unit_op_gpu.cu
  150. h_softmax_op.cc
  151. h_softmax_op.h
  152. half_float_ops.cc
  153. half_float_ops.cu
  154. half_float_ops.h
  155. if_op.cc
  156. if_op.h
  157. im2col_op.cc
  158. im2col_op.h
  159. im2col_op_gpu.cc
  160. index_hash_ops.cc
  161. index_hash_ops.h
  162. index_ops.cc
  163. instance_norm_gradient_op.cc
  164. instance_norm_op.cc
  165. instance_norm_op.cu
  166. instance_norm_op.h
  167. last_n_window_collector.cc
  168. layer_norm_op.cc
  169. layer_norm_op.cu
  170. layer_norm_op.h
  171. leaky_relu_op.cc
  172. leaky_relu_op.cu
  173. leaky_relu_op.h
  174. lengths_reducer_ops.cc
  175. lengths_reducer_ops.h
  176. lengths_reducer_rowwise_8bit_ops.cc
  177. lengths_reducer_rowwise_8bit_ops.h
  178. lengths_tile_op.cc
  179. lengths_tile_op.h
  180. lengths_top_k_op.cc
  181. lengths_top_k_op.h
  182. load_save_op.cc
  183. load_save_op.h
  184. load_save_op_gpu.cc
  185. local_response_normalization_op.cc
  186. local_response_normalization_op.cu
  187. local_response_normalization_op.h
  188. local_response_normalization_op_cudnn.cc
  189. log_op.cc
  190. log_op.cu
  191. logit_op.cc
  192. logit_op.cu
  193. logit_op.h
  194. loss_op.cc
  195. loss_op.cu
  196. loss_op.h
  197. lp_pool_op.cc
  198. lp_pool_op.cu
  199. lpnorm_op.cc
  200. lpnorm_op.h
  201. lstm_unit_op.cc
  202. lstm_unit_op.h
  203. lstm_unit_op_gpu.cu
  204. map_ops.cc
  205. map_ops.h
  206. margin_ranking_criterion_op.cc
  207. margin_ranking_criterion_op.cu
  208. margin_ranking_criterion_op.h
  209. math_ops.cc
  210. math_ops.cu
  211. math_ops.h
  212. matmul_op.cc
  213. matmul_op.h
  214. matmul_op_gpu.cc
  215. max_pool_with_index.cu
  216. max_pool_with_index.h
  217. mem_query_op.cu
  218. merge_id_lists_op.cc
  219. merge_id_lists_op.h
  220. multi_class_accuracy_op.cc
  221. multi_class_accuracy_op.cu
  222. multi_class_accuracy_op.h
  223. negative_op.cc
  224. negative_op.cu
  225. no_default_engine_op.h
  226. normalize_op.cc
  227. normalize_op.cu
  228. normalize_op.h
  229. one_hot_ops.cc
  230. one_hot_ops.cu
  231. one_hot_ops.h
  232. operator_fallback_gpu.h
  233. operator_fallback_gpu_test.cc
  234. order_switch_ops.cc
  235. order_switch_ops.cu
  236. order_switch_ops.h
  237. pack_rnn_sequence_op.cc
  238. pack_rnn_sequence_op.h
  239. pack_segments.cc
  240. pack_segments.h
  241. pack_segments_op_gpu.cc
  242. pad_op.cc
  243. pad_op.h
  244. pad_op_gpu.cu
  245. partition_ops.cc
  246. partition_ops.h
  247. perplexity_op.cc
  248. perplexity_op.cu
  249. perplexity_op.h
  250. piecewise_linear_transform_op.cc
  251. piecewise_linear_transform_op.cu
  252. piecewise_linear_transform_op.h
  253. pool_gradient_op.cc
  254. pool_op.cc
  255. pool_op.cu
  256. pool_op.h
  257. pool_op_cudnn.cu
  258. prefetch_op.h
  259. prelu_op.cc
  260. prelu_op.cu
  261. prelu_op.h
  262. prepend_dim_op.cc
  263. prepend_dim_op.h
  264. prepend_dim_op_gpu.cc
  265. rank_loss_op.cc
  266. rank_loss_op.h
  267. recurrent_network_blob_fetcher_op.cc
  268. recurrent_network_blob_fetcher_op.h
  269. recurrent_network_blob_fetcher_op_gpu.cc
  270. recurrent_network_executor.cc
  271. recurrent_network_executor.h
  272. recurrent_network_executor_gpu.cc
  273. recurrent_network_executor_gpu.h
  274. recurrent_network_executor_incl.h
  275. recurrent_network_op.cc
  276. recurrent_network_op.h
  277. recurrent_network_op_gpu.cu
  278. recurrent_op_cudnn.cc
  279. recurrent_op_cudnn.h
  280. reducer_functors.h
  281. reduction_front_back_ops.cc
  282. reduction_front_back_ops.cu
  283. reduction_front_back_ops.h
  284. reduction_ops.cc
  285. reduction_ops.cu
  286. reduction_ops.h
  287. relu_op.cc
  288. relu_op.cu
  289. relu_op.h
  290. relu_op_cudnn.cc
  291. relu_op_fp16.cu
  292. remove_data_blocks_op.cc
  293. remove_data_blocks_op.h
  294. replace_nan_op.cc
  295. replace_nan_op.h
  296. reservoir_sampling.cc
  297. reshape_op.cc
  298. reshape_op.h
  299. reshape_op_gpu.cc
  300. reshape_op_gpu_test.cc
  301. resize_op.cc
  302. resize_op.cu
  303. resize_op.h
  304. reverse_packed_segs_op.cc
  305. reverse_packed_segs_op.cu
  306. reverse_packed_segs_op.h
  307. rmac_regions_op.cc
  308. rmac_regions_op.cu
  309. rmac_regions_op.h
  310. roi_pool_op.cc
  311. roi_pool_op.cu
  312. roi_pool_op.h
  313. rowmul_op.cc
  314. rowmul_op.h
  315. scale_op.cc
  316. scale_op.h
  317. scale_op_gpu.cc
  318. segment_reduction_op.cc
  319. segment_reduction_op.h
  320. segment_reduction_op_gpu.cu
  321. sequence_ops.cc
  322. sequence_ops.h
  323. shape_op.cc
  324. shape_op.h
  325. shape_op_gpu.cc
  326. sigmoid_op.cc
  327. sigmoid_op.cu
  328. sin_op.cc
  329. sin_op.cu
  330. sinusoid_position_encoding_op.cc
  331. sinusoid_position_encoding_op.h
  332. slice_op.cc
  333. slice_op.cu
  334. slice_op.h
  335. softmax_op.cc
  336. softmax_op.h
  337. softmax_op_cudnn.cc
  338. softmax_ops.cu
  339. softmax_shared.cc
  340. softmax_shared.h
  341. softmax_with_loss_op.cc
  342. softmax_with_loss_op.h
  343. softplus_op.cc
  344. softplus_op.cu
  345. softplus_op.h
  346. softsign_op.cc
  347. softsign_op.cu
  348. space_batch_op.cc
  349. space_batch_op.h
  350. space_batch_op_gpu.cu
  351. sparse_to_dense_mask_op.cc
  352. sparse_to_dense_mask_op.h
  353. sparse_to_dense_op.cc
  354. sparse_to_dense_op.cu
  355. sparse_to_dense_op.h
  356. spatial_batch_norm_gradient_op.cc
  357. spatial_batch_norm_op.cc
  358. spatial_batch_norm_op.h
  359. spatial_batch_norm_op_cudnn.cc
  360. spatial_softmax_with_loss_op.cc
  361. spatial_softmax_with_loss_op.h
  362. square_root_divide_op.cc
  363. square_root_divide_op.h
  364. stats_ops.cc
  365. stop_gradient.cc
  366. stop_gradient.h
  367. stop_gradient_gpu.cc
  368. string_ops.cc
  369. string_ops.h
  370. string_ops_test.cc
  371. summarize_op.cc
  372. summarize_op.cu
  373. summarize_op.h
  374. tanh_op.cc
  375. tanh_op.cu
  376. tensor_protos_db_input.cc
  377. tensor_protos_db_input.h
  378. tensor_protos_db_input_gpu.cc
  379. text_file_reader.cc
  380. text_file_reader_utils.cc
  381. text_file_reader_utils.h
  382. text_file_reader_utils_test.cc
  383. tile_op.cc
  384. tile_op.cu
  385. tile_op.h
  386. top_k.cc
  387. top_k.cu
  388. top_k.h
  389. top_k_heap_selection.cuh
  390. top_k_radix_selection.cuh
  391. transpose_op.cc
  392. transpose_op.cu
  393. transpose_op.h
  394. transpose_op_cudnn.cc
  395. tt_linear_op.cc
  396. tt_linear_op.h
  397. utility_ops.cc
  398. utility_ops.cu
  399. utility_ops.h
  400. utility_ops_gpu.cc
  401. utility_ops_gpu_test.cc
  402. utility_ops_test.cc
  403. weighted_sample_op.cc
  404. weighted_sample_op.h
  405. while_op.cc
  406. while_op.h
  407. workspace_ops.cc
  408. zero_gradient_op.cc
  409. zero_gradient_op.h
  410. zero_gradient_op_gpu.cc