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