Google Git
Sign in
android / platform / external / pytorch / b294aadc66 / . / caffe2 / operators
tree: f642baa9ad7ed76fc004220cb81ad813170c2326 [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_mul_op.cc
  70. elementwise_op.cc
  71. elementwise_op.cu
  72. elementwise_op.h
  73. elementwise_op_gpu_test.cc
  74. elementwise_op_schema.cc
  75. elementwise_op_test.cc
  76. elementwise_op_test.h
  77. elu_op.cc
  78. elu_op.h
  79. exp_op.cc
  80. exp_op.cu
  81. extend_tensor_op.cc
  82. feed_blob_op.cc
  83. feed_blob_op.h
  84. filler_op.cc
  85. filler_op.cu
  86. filler_op.h
  87. find_duplicate_elements_op.cc
  88. find_duplicate_elements_op.h
  89. free_op.cc
  90. free_op.h
  91. free_op_gpu.cc
  92. fully_connected_op.cc
  93. fully_connected_op.h
  94. fully_connected_op_gpu.cc
  95. fully_connected_op_gpu_test.cc
  96. fully_connected_op_test.cc
  97. given_tensor_fill_op.cc
  98. given_tensor_fill_op.cu
  99. given_tensor_fill_op.h
  100. h_softmax_op.cc
  101. h_softmax_op.h
  102. half_float_ops.cc
  103. half_float_ops.cu
  104. half_float_ops.h
  105. im2col_op.cc
  106. im2col_op.h
  107. im2col_op_gpu.cc
  108. index_ops.cc
  109. instance_norm_gradient_op.cc
  110. instance_norm_op.cc
  111. instance_norm_op.cu
  112. instance_norm_op.h
  113. last_n_window_collector.cc
  114. leaky_relu_op.cc
  115. leaky_relu_op.cu
  116. leaky_relu_op.h
  117. lengths_tile_op.cc
  118. lengths_tile_op.h
  119. load_save_op.cc
  120. load_save_op.h
  121. load_save_op_gpu.cc
  122. local_response_normalization_op.cc
  123. local_response_normalization_op.cu
  124. local_response_normalization_op.h
  125. logit_op.cc
  126. loss_op.cc
  127. loss_op.cu
  128. loss_op.h
  129. lp_pool_op.cc
  130. lp_pool_op.cu
  131. lstm_unit_op.cc
  132. lstm_unit_op.h
  133. lstm_unit_op_gpu.cu
  134. margin_ranking_criterion_op.cc
  135. margin_ranking_criterion_op.cu
  136. margin_ranking_criterion_op.h
  137. math_ops.cc
  138. math_ops.cu
  139. math_ops.h
  140. matmul_op.cc
  141. matmul_op.h
  142. matmul_op_gpu.cc
  143. mem_query_op.cu
  144. metrics_ops.cc
  145. metrics_ops.h
  146. multi_class_accuracy_op.cc
  147. multi_class_accuracy_op.cu
  148. multi_class_accuracy_op.h
  149. negative_op.cc
  150. negative_op.cu
  151. no_default_engine_op.h
  152. normalize_op.cc
  153. normalize_op.h
  154. one_hot_ops.cc
  155. one_hot_ops.h
  156. operator_fallback_gpu.h
  157. operator_fallback_gpu_test.cc
  158. order_switch_ops.cc
  159. order_switch_ops.cu
  160. order_switch_ops.h
  161. pack_segments.cc
  162. pack_segments.h
  163. pack_segments_op_gpu.cc
  164. pad_op.cc
  165. pad_op.h
  166. pad_op_gpu.cu
  167. partition_ops.cc
  168. partition_ops.h
  169. perplexity_op.cc
  170. perplexity_op.cu
  171. perplexity_op.h
  172. piecewise_linear_transform_op.cc
  173. piecewise_linear_transform_op.h
  174. pool_gradient_op.cc
  175. pool_op.cc
  176. pool_op.cu
  177. pool_op.h
  178. pool_op_cudnn.cc
  179. prefetch_op.h
  180. prelu_op.cc
  181. prelu_op.h
  182. rank_loss_op.cc
  183. rank_loss_op.h
  184. recurrent_network_op.cc
  185. recurrent_network_op.h
  186. recurrent_network_op_gpu.cc
  187. recurrent_op_cudnn.cc
  188. recurrent_op_cudnn.h
  189. reducer_functors.h
  190. reduction_ops.cc
  191. reduction_ops.cu
  192. reduction_ops.h
  193. relu_op.cc
  194. relu_op.cu
  195. relu_op.h
  196. relu_op_cudnn.cc
  197. relu_op_fp16.cu
  198. remove_data_blocks_op.cc
  199. remove_data_blocks_op.h
  200. replace_nan_op.cc
  201. replace_nan_op.h
  202. reshape_op.cc
  203. reshape_op.h
  204. reshape_op_gpu.cc
  205. reshape_op_gpu_test.cc
  206. resize_op.cc
  207. resize_op.h
  208. reverse_packed_segs_op.cc
  209. reverse_packed_segs_op.cu
  210. reverse_packed_segs_op.h
  211. roi_pool_op.cc
  212. roi_pool_op.cu
  213. roi_pool_op.h
  214. rowmul_op.cc
  215. rowmul_op.h
  216. scale_op.cc
  217. scale_op.h
  218. scale_op_gpu.cc
  219. segment_reduction_op.cc
  220. segment_reduction_op.h
  221. segment_reduction_op_gpu.cu
  222. sequence_ops.cc
  223. sequence_ops.h
  224. sigmoid_op.cc
  225. sigmoid_op.cu
  226. softmax_op.cc
  227. softmax_op.h
  228. softmax_op_cudnn.cc
  229. softmax_ops.cu
  230. softmax_shared.cc
  231. softmax_shared.h
  232. softmax_with_loss_op.cc
  233. softmax_with_loss_op.h
  234. softsign_op.cc
  235. softsign_op.cu
  236. space_batch_op.cc
  237. space_batch_op.h
  238. space_batch_op_gpu.cu
  239. sparse_to_dense_mask_op.cc
  240. sparse_to_dense_op.cc
  241. sparse_to_dense_op.h
  242. spatial_batch_norm_gradient_op.cc
  243. spatial_batch_norm_op.cc
  244. spatial_batch_norm_op.h
  245. spatial_batch_norm_op_cudnn.cc
  246. square_root_divide_op.cc
  247. square_root_divide_op.h
  248. stats_ops.cc
  249. stop_gradient.cc
  250. stop_gradient.h
  251. stop_gradient_gpu.cc
  252. string_ops.cc
  253. string_ops.h
  254. string_ops_test.cc
  255. summarize_op.cc
  256. summarize_op.cu
  257. summarize_op.h
  258. tanh_op.cc
  259. tanh_op.cu
  260. tensor_protos_db_input.cc
  261. tensor_protos_db_input.h
  262. tensor_protos_db_input_gpu.cc
  263. text_file_reader.cc
  264. text_file_reader_utils.cc
  265. text_file_reader_utils.h
  266. text_file_reader_utils_test.cc
  267. tile_op.cc
  268. tile_op.h
  269. tile_op_gpu.cc
  270. top_k.cc
  271. top_k.h
  272. top_k_gpu.cc
  273. transpose_op.cc
  274. transpose_op.cu
  275. transpose_op.h
  276. transpose_op_cudnn.cc
  277. tt_linear_op.cc
  278. tt_linear_op.h
  279. utility_ops.cc
  280. utility_ops.cu
  281. utility_ops.h
  282. utility_ops_gpu.cc
  283. utility_ops_gpu_test.cc
  284. utility_ops_test.cc
  285. workspace_ops.cc