Google Git
Sign in
android / platform / external / pytorch / 965a7daf9b / . / caffe2 / operators
tree: 5341d84c8ffe14a57401bdfe21fb609d4ef3e6d9 [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.h
  10. batch_matmul_op_gpu.cc
  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.h
  112. load_save_op.cc
  113. load_save_op.h
  114. load_save_op_gpu.cc
  115. local_response_normalization_op.cc
  116. local_response_normalization_op.cu
  117. local_response_normalization_op.h
  118. loss_op.cc
  119. loss_op.cu
  120. loss_op.h
  121. lp_pool_op.cc
  122. lp_pool_op.cu
  123. lstm_unit_op.cc
  124. lstm_unit_op.h
  125. lstm_unit_op_gpu.cu
  126. margin_ranking_criterion_op.cc
  127. margin_ranking_criterion_op.cu
  128. margin_ranking_criterion_op.h
  129. math_ops.cc
  130. math_ops.cu
  131. math_ops.h
  132. matmul_op.cc
  133. matmul_op.h
  134. matmul_op_gpu.cc
  135. metrics_ops.cc
  136. metrics_ops.h
  137. multi_class_accuracy_op.cc
  138. multi_class_accuracy_op.cu
  139. multi_class_accuracy_op.h
  140. negative_op.cc
  141. negative_op.cu
  142. no_default_engine_op.h
  143. normalize_op.cc
  144. normalize_op.h
  145. one_hot_ops.cc
  146. operator_fallback_gpu.h
  147. operator_fallback_gpu_test.cc
  148. order_switch_ops.cc
  149. order_switch_ops.cu
  150. order_switch_ops.h
  151. pack_segments.cc
  152. pad_op.cc
  153. pad_op.h
  154. pad_op_gpu.cu
  155. partition_ops.cc
  156. partition_ops.h
  157. perplexity_op.cc
  158. perplexity_op.cu
  159. perplexity_op.h
  160. piecewise_linear_transform_op.cc
  161. piecewise_linear_transform_op.h
  162. pool_gradient_op.cc
  163. pool_op.cc
  164. pool_op.cu
  165. pool_op.h
  166. pool_op_cudnn.cc
  167. prefetch_op.h
  168. prelu_op.cc
  169. prelu_op.h
  170. rank_loss_op.cc
  171. rank_loss_op.h
  172. recurrent_network_op.cc
  173. recurrent_network_op.h
  174. recurrent_network_op_gpu.cc
  175. recurrent_op_cudnn.cc
  176. recurrent_op_cudnn.h
  177. reducer_functors.h
  178. relu_op.cc
  179. relu_op.cu
  180. relu_op.h
  181. relu_op_cudnn.cc
  182. relu_op_fp16.cu
  183. remove_data_blocks_op.cc
  184. remove_data_blocks_op.h
  185. reverse_packed_segs_op.cc
  186. reverse_packed_segs_op.cu
  187. reverse_packed_segs_op.h
  188. roi_pool_op.cc
  189. roi_pool_op.cu
  190. roi_pool_op.h
  191. rowmul_op.cc
  192. rowmul_op.h
  193. scale_op.cc
  194. scale_op.h
  195. scale_op_gpu.cc
  196. segment_reduction_op.cc
  197. segment_reduction_op.h
  198. segment_reduction_op_gpu.cu
  199. sequence_ops.cc
  200. sigmoid_op.cc
  201. sigmoid_op.cu
  202. softmax_op.cc
  203. softmax_op.cu
  204. softmax_op.h
  205. softmax_op_cudnn.cc
  206. softmax_shared.cc
  207. softmax_shared.h
  208. softmax_with_loss_op.cc
  209. softmax_with_loss_op.cu
  210. softmax_with_loss_op.h
  211. softsign_op.cc
  212. softsign_op.cu
  213. space_batch_op.cc
  214. space_batch_op.h
  215. space_batch_op_gpu.cu
  216. sparse_to_dense_mask_op.cc
  217. sparse_to_dense_op.cc
  218. sparse_to_dense_op.h
  219. spatial_batch_norm_gradient_op.cc
  220. spatial_batch_norm_op.cc
  221. spatial_batch_norm_op.h
  222. spatial_batch_norm_op_cudnn.cc
  223. square_root_divide_op.cc
  224. square_root_divide_op.h
  225. stats_ops.cc
  226. stop_gradient.cc
  227. stop_gradient.h
  228. stop_gradient_gpu.cc
  229. string_ops.cc
  230. string_ops.h
  231. string_ops_test.cc
  232. summarize_op.cc
  233. summarize_op.cu
  234. summarize_op.h
  235. tanh_op.cc
  236. tanh_op.cu
  237. tensor_protos_db_input.cc
  238. tensor_protos_db_input.h
  239. tensor_protos_db_input_gpu.cc
  240. text_file_reader.cc
  241. text_file_reader_utils.cc
  242. text_file_reader_utils.h
  243. text_file_reader_utils_test.cc
  244. tile_op.cc
  245. tile_op.h
  246. tile_op_gpu.cc
  247. transpose_op.cc
  248. transpose_op.cu
  249. transpose_op.h
  250. tt_linear_op.cc
  251. tt_linear_op.h
  252. utility_ops.cc
  253. utility_ops.cu
  254. utility_ops.h
  255. utility_ops_gpu.cc
  256. utility_ops_gpu_test.cc
  257. utility_ops_test.cc
  258. workspace_ops.cc