Google Git
Sign in
android / platform / external / pytorch / 6c67a753c7 / . / caffe2 / operators
tree: d6e1275ae9b0300720a406c4e3e0b624d0abadec [path history] [tgz]
  1. accumulate_op.cc
  2. accumulate_op.cu
  3. accumulate_op.h
  4. accuracy_op.cc
  5. accuracy_op.cu
  6. accuracy_op.h
  7. apmeter_op.cc
  8. apmeter_op.h
  9. atomic_ops.cc
  10. batch_box_cox_op.cc
  11. batch_box_cox_op.h
  12. batch_matmul_op.cc
  13. batch_matmul_op.cu
  14. batch_matmul_op.h
  15. boolean_mask_ops.cc
  16. boolean_unmask_ops.cc
  17. boolean_unmask_ops_test.cc
  18. cast_op.cc
  19. cast_op.cu
  20. cast_op.h
  21. clip_op.cc
  22. clip_op.cu
  23. clip_op.h
  24. CMakeLists.txt
  25. communicator_op.cc
  26. communicator_op_gpu.cc
  27. concat_split_op.cc
  28. concat_split_op.h
  29. concat_split_op_gpu.cc
  30. conv_gradient_op.cc
  31. conv_op.cc
  32. conv_op.h
  33. conv_op_cache_cudnn.cc
  34. conv_op_cache_cudnn.h
  35. conv_op_cache_cudnn_test.cc
  36. conv_op_cudnn.cc
  37. conv_op_eigen.cc
  38. conv_op_gpu.cc
  39. conv_op_impl.h
  40. conv_op_shared.cc
  41. conv_op_shared.h
  42. conv_op_shared_gpu.cc
  43. conv_pool_op_base.h
  44. conv_transpose_gradient_op.cc
  45. conv_transpose_op.cc
  46. conv_transpose_op.h
  47. conv_transpose_op_cudnn.cc
  48. conv_transpose_op_gpu.cc
  49. conv_transpose_op_impl.h
  50. conv_transpose_op_mobile.cc
  51. conv_transpose_op_mobile.h
  52. conv_transpose_op_mobile_impl.h
  53. conv_transpose_unpool_op_base.h
  54. cosine_embedding_criterion_op.cc
  55. cosine_embedding_criterion_op.cu
  56. cosine_embedding_criterion_op.h
  57. counter_ops.cc
  58. counter_ops.h
  59. counter_ops_gpu.cc
  60. cross_entropy_op.cc
  61. cross_entropy_op.cu
  62. cross_entropy_op.h
  63. dataset_ops.cc
  64. distance_op.cc
  65. distance_op.cu
  66. distance_op.h
  67. dropout_op.cc
  68. dropout_op.cu
  69. dropout_op.h
  70. dropout_op_cudnn.cc
  71. elementwise_add_op.cc
  72. elementwise_div_op.cc
  73. elementwise_linear_op.cc
  74. elementwise_linear_op.cu
  75. elementwise_linear_op.h
  76. elementwise_logical_ops.cc
  77. elementwise_logical_ops.h
  78. elementwise_mul_op.cc
  79. elementwise_op.cc
  80. elementwise_op.cu
  81. elementwise_op.h
  82. elementwise_op_gpu_test.cc
  83. elementwise_op_schema.cc
  84. elementwise_op_test.cc
  85. elementwise_op_test.h
  86. elementwise_sub_op.cc
  87. elu_op.cc
  88. elu_op.cu
  89. elu_op.h
  90. exp_op.cc
  91. exp_op.cu
  92. extend_tensor_op.cc
  93. feed_blob_op.cc
  94. feed_blob_op.h
  95. filler_op.cc
  96. filler_op.cu
  97. filler_op.h
  98. find_duplicate_elements_op.cc
  99. find_duplicate_elements_op.h
  100. find_op.cc
  101. find_op.cu
  102. find_op.h
  103. free_op.cc
  104. free_op.h
  105. free_op_gpu.cc
  106. fully_connected_op.cc
  107. fully_connected_op.h
  108. fully_connected_op_gpu.cc
  109. fully_connected_op_gpu_test.cc
  110. fully_connected_op_test.cc
  111. given_tensor_fill_op.cc
  112. given_tensor_fill_op.cu
  113. given_tensor_fill_op.h
  114. h_softmax_op.cc
  115. h_softmax_op.h
  116. half_float_ops.cc
  117. half_float_ops.cu
  118. half_float_ops.h
  119. im2col_op.cc
  120. im2col_op.h
  121. im2col_op_gpu.cc
  122. index_ops.cc
  123. instance_norm_gradient_op.cc
  124. instance_norm_op.cc
  125. instance_norm_op.cu
  126. instance_norm_op.h
  127. last_n_window_collector.cc
  128. leaky_relu_op.cc
  129. leaky_relu_op.cu
  130. leaky_relu_op.h
  131. lengths_tile_op.cc
  132. lengths_tile_op.h
  133. load_save_op.cc
  134. load_save_op.h
  135. load_save_op_gpu.cc
  136. local_response_normalization_op.cc
  137. local_response_normalization_op.cu
  138. local_response_normalization_op.h
  139. local_response_normalization_op_cudnn.cc
  140. log_op.cc
  141. log_op.cu
  142. logit_op.cc
  143. loss_op.cc
  144. loss_op.cu
  145. loss_op.h
  146. lp_pool_op.cc
  147. lp_pool_op.cu
  148. lstm_unit_op.cc
  149. lstm_unit_op.h
  150. lstm_unit_op_gpu.cu
  151. map_ops.cc
  152. map_ops.h
  153. margin_ranking_criterion_op.cc
  154. margin_ranking_criterion_op.cu
  155. margin_ranking_criterion_op.h
  156. math_ops.cc
  157. math_ops.cu
  158. math_ops.h
  159. matmul_op.cc
  160. matmul_op.h
  161. matmul_op_gpu.cc
  162. max_pool_with_index.cu
  163. max_pool_with_index.h
  164. mem_query_op.cu
  165. metrics_ops.cc
  166. metrics_ops.h
  167. multi_class_accuracy_op.cc
  168. multi_class_accuracy_op.cu
  169. multi_class_accuracy_op.h
  170. negative_op.cc
  171. negative_op.cu
  172. no_default_engine_op.h
  173. normalize_op.cc
  174. normalize_op.cu
  175. normalize_op.h
  176. one_hot_ops.cc
  177. one_hot_ops.h
  178. operator_fallback_gpu.h
  179. operator_fallback_gpu_test.cc
  180. order_switch_ops.cc
  181. order_switch_ops.cu
  182. order_switch_ops.h
  183. pack_segments.cc
  184. pack_segments.h
  185. pack_segments_op_gpu.cc
  186. pad_op.cc
  187. pad_op.h
  188. pad_op_gpu.cu
  189. partition_ops.cc
  190. partition_ops.h
  191. perplexity_op.cc
  192. perplexity_op.cu
  193. perplexity_op.h
  194. piecewise_linear_transform_op.cc
  195. piecewise_linear_transform_op.h
  196. pool_gradient_op.cc
  197. pool_op.cc
  198. pool_op.cu
  199. pool_op.h
  200. pool_op_cudnn.cc
  201. prefetch_op.h
  202. prelu_op.cc
  203. prelu_op.cu
  204. prelu_op.h
  205. rank_loss_op.cc
  206. rank_loss_op.h
  207. recurrent_network_executor.h
  208. recurrent_network_op.cc
  209. recurrent_network_op.h
  210. recurrent_network_op_gpu.cc
  211. recurrent_op_cudnn.cc
  212. recurrent_op_cudnn.h
  213. reducer_functors.h
  214. reduction_ops.cc
  215. reduction_ops.cu
  216. reduction_ops.h
  217. relu_op.cc
  218. relu_op.cu
  219. relu_op.h
  220. relu_op_cudnn.cc
  221. relu_op_fp16.cu
  222. remove_data_blocks_op.cc
  223. remove_data_blocks_op.h
  224. replace_nan_op.cc
  225. replace_nan_op.h
  226. reshape_op.cc
  227. reshape_op.h
  228. reshape_op_gpu.cc
  229. reshape_op_gpu_test.cc
  230. resize_op.cc
  231. resize_op.cu
  232. resize_op.h
  233. reverse_packed_segs_op.cc
  234. reverse_packed_segs_op.cu
  235. reverse_packed_segs_op.h
  236. roi_pool_op.cc
  237. roi_pool_op.cu
  238. roi_pool_op.h
  239. rowmul_op.cc
  240. rowmul_op.h
  241. scale_op.cc
  242. scale_op.h
  243. scale_op_gpu.cc
  244. segment_reduction_op.cc
  245. segment_reduction_op.h
  246. segment_reduction_op_gpu.cu
  247. sequence_ops.cc
  248. sequence_ops.h
  249. sigmoid_op.cc
  250. sigmoid_op.cu
  251. softmax_op.cc
  252. softmax_op.h
  253. softmax_op_cudnn.cc
  254. softmax_ops.cu
  255. softmax_shared.cc
  256. softmax_shared.h
  257. softmax_with_loss_op.cc
  258. softmax_with_loss_op.h
  259. softplus_op.cc
  260. softplus_op.cu
  261. softplus_op.h
  262. softsign_op.cc
  263. softsign_op.cu
  264. space_batch_op.cc
  265. space_batch_op.h
  266. space_batch_op_gpu.cu
  267. sparse_to_dense_mask_op.cc
  268. sparse_to_dense_op.cc
  269. sparse_to_dense_op.h
  270. spatial_batch_norm_gradient_op.cc
  271. spatial_batch_norm_op.cc
  272. spatial_batch_norm_op.h
  273. spatial_batch_norm_op_cudnn.cc
  274. spatial_softmax_with_loss_op.cc
  275. spatial_softmax_with_loss_op.h
  276. square_root_divide_op.cc
  277. square_root_divide_op.h
  278. stats_ops.cc
  279. stop_gradient.cc
  280. stop_gradient.h
  281. stop_gradient_gpu.cc
  282. string_ops.cc
  283. string_ops.h
  284. string_ops_test.cc
  285. summarize_op.cc
  286. summarize_op.cu
  287. summarize_op.h
  288. tanh_op.cc
  289. tanh_op.cu
  290. tensor_protos_db_input.cc
  291. tensor_protos_db_input.h
  292. tensor_protos_db_input_gpu.cc
  293. text_file_reader.cc
  294. text_file_reader_utils.cc
  295. text_file_reader_utils.h
  296. text_file_reader_utils_test.cc
  297. tile_op.cc
  298. tile_op.cu
  299. tile_op.h
  300. top_k.cc
  301. top_k.cu
  302. top_k.h
  303. top_k_heap_selection.cuh
  304. top_k_radix_selection.cuh
  305. transpose_op.cc
  306. transpose_op.cu
  307. transpose_op.h
  308. transpose_op_cudnn.cc
  309. tt_linear_op.cc
  310. tt_linear_op.h
  311. utility_ops.cc
  312. utility_ops.cu
  313. utility_ops.h
  314. utility_ops_gpu.cc
  315. utility_ops_gpu_test.cc
  316. utility_ops_test.cc
  317. workspace_ops.cc
  318. zero_gradient_op.cc
  319. zero_gradient_op.h
  320. zero_gradient_op_gpu.cc