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