Google Git
Sign in
android / platform / external / pytorch / e9581e47a2 / . / caffe2 / operators
tree: ba09fc61c4ffb4a12951c8a953eada9af7db6ab8 [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. cross_entropy_op.cc
  78. cross_entropy_op.cu
  79. cross_entropy_op.h
  80. dataset_ops.cc
  81. dataset_ops.h
  82. distance_op.cc
  83. distance_op.cu
  84. distance_op.h
  85. do_op.cc
  86. do_op.h
  87. dropout_op.cc
  88. dropout_op.cu
  89. dropout_op.h
  90. dropout_op_cudnn.cc
  91. elementwise_add_op.cc
  92. elementwise_div_op.cc
  93. elementwise_linear_op.cc
  94. elementwise_linear_op.cu
  95. elementwise_linear_op.h
  96. elementwise_logical_ops.cc
  97. elementwise_logical_ops.h
  98. elementwise_mul_op.cc
  99. elementwise_op.cc
  100. elementwise_op.cu
  101. elementwise_op.h
  102. elementwise_op_gpu_test.cc
  103. elementwise_op_schema.cc
  104. elementwise_op_test.cc
  105. elementwise_op_test.h
  106. elementwise_sub_op.cc
  107. elementwise_sum_op.cc
  108. elu_op.cc
  109. elu_op.cu
  110. elu_op.h
  111. exp_op.cc
  112. exp_op.cu
  113. extend_tensor_op.cc
  114. feed_blob_op.cc
  115. feed_blob_op.h
  116. filler_op.cc
  117. filler_op.cu
  118. filler_op.h
  119. find_duplicate_elements_op.cc
  120. find_duplicate_elements_op.h
  121. find_op.cc
  122. find_op.cu
  123. find_op.h
  124. free_op.cc
  125. free_op.h
  126. free_op_gpu.cc
  127. fully_connected_op.cc
  128. fully_connected_op.h
  129. fully_connected_op_gpu.cc
  130. fully_connected_op_gpu_test.cc
  131. fully_connected_op_test.cc
  132. given_tensor_fill_op.cc
  133. given_tensor_fill_op.cu
  134. given_tensor_fill_op.h
  135. gru_unit_op.cc
  136. gru_unit_op.h
  137. gru_unit_op_gpu.cu
  138. h_softmax_op.cc
  139. h_softmax_op.h
  140. half_float_ops.cc
  141. half_float_ops.cu
  142. half_float_ops.h
  143. if_op.cc
  144. if_op.h
  145. im2col_op.cc
  146. im2col_op.h
  147. im2col_op_gpu.cc
  148. index_hash_ops.cc
  149. index_hash_ops.h
  150. index_ops.cc
  151. instance_norm_gradient_op.cc
  152. instance_norm_op.cc
  153. instance_norm_op.cu
  154. instance_norm_op.h
  155. last_n_window_collector.cc
  156. layer_norm_op.cc
  157. layer_norm_op.cu
  158. layer_norm_op.h
  159. leaky_relu_op.cc
  160. leaky_relu_op.cu
  161. leaky_relu_op.h
  162. lengths_reducer_ops.cc
  163. lengths_reducer_ops.h
  164. lengths_reducer_rowwise_8bit_ops.cc
  165. lengths_reducer_rowwise_8bit_ops.h
  166. lengths_tile_op.cc
  167. lengths_tile_op.h
  168. lengths_top_k_op.cc
  169. lengths_top_k_op.h
  170. load_save_op.cc
  171. load_save_op.h
  172. load_save_op_gpu.cc
  173. local_response_normalization_op.cc
  174. local_response_normalization_op.cu
  175. local_response_normalization_op.h
  176. local_response_normalization_op_cudnn.cc
  177. log_op.cc
  178. log_op.cu
  179. logit_op.cc
  180. loss_op.cc
  181. loss_op.cu
  182. loss_op.h
  183. lp_pool_op.cc
  184. lp_pool_op.cu
  185. lpnorm_op.cc
  186. lpnorm_op.h
  187. lstm_unit_op.cc
  188. lstm_unit_op.h
  189. lstm_unit_op_gpu.cu
  190. map_ops.cc
  191. map_ops.h
  192. margin_ranking_criterion_op.cc
  193. margin_ranking_criterion_op.cu
  194. margin_ranking_criterion_op.h
  195. math_ops.cc
  196. math_ops.cu
  197. math_ops.h
  198. matmul_op.cc
  199. matmul_op.h
  200. matmul_op_gpu.cc
  201. max_pool_with_index.cu
  202. max_pool_with_index.h
  203. mem_query_op.cu
  204. merge_id_lists_op.cc
  205. merge_id_lists_op.h
  206. multi_class_accuracy_op.cc
  207. multi_class_accuracy_op.cu
  208. multi_class_accuracy_op.h
  209. negative_op.cc
  210. negative_op.cu
  211. no_default_engine_op.h
  212. normalize_op.cc
  213. normalize_op.cu
  214. normalize_op.h
  215. one_hot_ops.cc
  216. one_hot_ops.cu
  217. one_hot_ops.h
  218. operator_fallback_gpu.h
  219. operator_fallback_gpu_test.cc
  220. order_switch_ops.cc
  221. order_switch_ops.cu
  222. order_switch_ops.h
  223. pack_rnn_sequence_op.cc
  224. pack_rnn_sequence_op.h
  225. pack_segments.cc
  226. pack_segments.h
  227. pack_segments_op_gpu.cc
  228. pad_op.cc
  229. pad_op.h
  230. pad_op_gpu.cu
  231. partition_ops.cc
  232. partition_ops.h
  233. perplexity_op.cc
  234. perplexity_op.cu
  235. perplexity_op.h
  236. piecewise_linear_transform_op.cc
  237. piecewise_linear_transform_op.cu
  238. piecewise_linear_transform_op.h
  239. pool_gradient_op.cc
  240. pool_op.cc
  241. pool_op.cu
  242. pool_op.h
  243. pool_op_cudnn.cu
  244. prefetch_op.h
  245. prelu_op.cc
  246. prelu_op.cu
  247. prelu_op.h
  248. prepend_dim_op.cc
  249. prepend_dim_op.h
  250. prepend_dim_op_gpu.cc
  251. rank_loss_op.cc
  252. rank_loss_op.h
  253. recurrent_network_blob_fetcher_op.cc
  254. recurrent_network_blob_fetcher_op.h
  255. recurrent_network_blob_fetcher_op_gpu.cc
  256. recurrent_network_executor.cc
  257. recurrent_network_executor.h
  258. recurrent_network_executor_gpu.cc
  259. recurrent_network_executor_gpu.h
  260. recurrent_network_executor_incl.h
  261. recurrent_network_op.cc
  262. recurrent_network_op.h
  263. recurrent_network_op_gpu.cc
  264. recurrent_op_cudnn.cc
  265. recurrent_op_cudnn.h
  266. reducer_functors.h
  267. reduction_ops.cc
  268. reduction_ops.cu
  269. reduction_ops.h
  270. relu_op.cc
  271. relu_op.cu
  272. relu_op.h
  273. relu_op_cudnn.cc
  274. relu_op_fp16.cu
  275. remove_data_blocks_op.cc
  276. remove_data_blocks_op.h
  277. replace_nan_op.cc
  278. replace_nan_op.h
  279. reservoir_sampling.cc
  280. reshape_op.cc
  281. reshape_op.h
  282. reshape_op_gpu.cc
  283. reshape_op_gpu_test.cc
  284. resize_op.cc
  285. resize_op.cu
  286. resize_op.h
  287. reverse_packed_segs_op.cc
  288. reverse_packed_segs_op.cu
  289. reverse_packed_segs_op.h
  290. rmac_regions_op.cc
  291. rmac_regions_op.cu
  292. rmac_regions_op.h
  293. roi_pool_op.cc
  294. roi_pool_op.cu
  295. roi_pool_op.h
  296. rowmul_op.cc
  297. rowmul_op.h
  298. scale_op.cc
  299. scale_op.h
  300. scale_op_gpu.cc
  301. segment_reduction_op.cc
  302. segment_reduction_op.h
  303. segment_reduction_op_gpu.cu
  304. sequence_ops.cc
  305. sequence_ops.h
  306. shape_op.cc
  307. shape_op.h
  308. shape_op_gpu.cc
  309. sigmoid_op.cc
  310. sigmoid_op.cu
  311. sin_op.cc
  312. sin_op.cu
  313. sinusoid_position_encoding_op.cc
  314. sinusoid_position_encoding_op.h
  315. slice_op.cc
  316. slice_op.cu
  317. slice_op.h
  318. softmax_op.cc
  319. softmax_op.h
  320. softmax_op_cudnn.cc
  321. softmax_ops.cu
  322. softmax_shared.cc
  323. softmax_shared.h
  324. softmax_with_loss_op.cc
  325. softmax_with_loss_op.h
  326. softplus_op.cc
  327. softplus_op.cu
  328. softplus_op.h
  329. softsign_op.cc
  330. softsign_op.cu
  331. space_batch_op.cc
  332. space_batch_op.h
  333. space_batch_op_gpu.cu
  334. sparse_to_dense_mask_op.cc
  335. sparse_to_dense_mask_op.h
  336. sparse_to_dense_op.cc
  337. sparse_to_dense_op.cu
  338. sparse_to_dense_op.h
  339. spatial_batch_norm_gradient_op.cc
  340. spatial_batch_norm_op.cc
  341. spatial_batch_norm_op.h
  342. spatial_batch_norm_op_cudnn.cc
  343. spatial_softmax_with_loss_op.cc
  344. spatial_softmax_with_loss_op.h
  345. square_root_divide_op.cc
  346. square_root_divide_op.h
  347. stats_ops.cc
  348. stop_gradient.cc
  349. stop_gradient.h
  350. stop_gradient_gpu.cc
  351. string_ops.cc
  352. string_ops.h
  353. string_ops_test.cc
  354. summarize_op.cc
  355. summarize_op.cu
  356. summarize_op.h
  357. tanh_op.cc
  358. tanh_op.cu
  359. tensor_protos_db_input.cc
  360. tensor_protos_db_input.h
  361. tensor_protos_db_input_gpu.cc
  362. text_file_reader.cc
  363. text_file_reader_utils.cc
  364. text_file_reader_utils.h
  365. text_file_reader_utils_test.cc
  366. tile_op.cc
  367. tile_op.cu
  368. tile_op.h
  369. top_k.cc
  370. top_k.cu
  371. top_k.h
  372. top_k_heap_selection.cuh
  373. top_k_radix_selection.cuh
  374. transpose_op.cc
  375. transpose_op.cu
  376. transpose_op.h
  377. transpose_op_cudnn.cc
  378. tt_linear_op.cc
  379. tt_linear_op.h
  380. utility_ops.cc
  381. utility_ops.cu
  382. utility_ops.h
  383. utility_ops_gpu.cc
  384. utility_ops_gpu_test.cc
  385. utility_ops_test.cc
  386. while_op.cc
  387. while_op.h
  388. workspace_ops.cc
  389. zero_gradient_op.cc
  390. zero_gradient_op.h
  391. zero_gradient_op_gpu.cc