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