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