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