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