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