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