Google Git
Sign in
android / platform / external / pytorch / b51e0ec0c2 / . / caffe2 / operators
tree: 7374849b6b9b74b5d0e916169941bbfb0348e957 [path history] [tgz]
  1. abs_op.cc
  2. abs_op.cu
  3. accumulate_op.cc
  4. accumulate_op.cu
  5. accumulate_op.h
  6. accuracy_op.cc
  7. accuracy_op.cu
  8. accuracy_op.h
  9. apmeter_op.cc
  10. apmeter_op.h
  11. atomic_ops.cc
  12. batch_box_cox_op.cc
  13. batch_box_cox_op.h
  14. batch_matmul_op.cc
  15. batch_matmul_op.cu
  16. batch_matmul_op.h
  17. boolean_mask_ops.cc
  18. boolean_mask_ops.cu
  19. boolean_mask_ops.h
  20. boolean_unmask_ops.cc
  21. boolean_unmask_ops_test.cc
  22. cast_op.cc
  23. cast_op.cu
  24. cast_op.h
  25. clip_op.cc
  26. clip_op.cu
  27. clip_op.h
  28. CMakeLists.txt
  29. communicator_op.cc
  30. communicator_op_gpu.cc
  31. concat_split_op.cc
  32. concat_split_op.h
  33. concat_split_op_gpu.cc
  34. conv_gradient_op.cc
  35. conv_op.cc
  36. conv_op.h
  37. conv_op_cache_cudnn.cc
  38. conv_op_cache_cudnn.h
  39. conv_op_cache_cudnn_test.cc
  40. conv_op_cudnn.cc
  41. conv_op_eigen.cc
  42. conv_op_gpu.cc
  43. conv_op_impl.h
  44. conv_op_shared.cc
  45. conv_op_shared.h
  46. conv_op_shared_gpu.cc
  47. conv_pool_op_base.h
  48. conv_transpose_gradient_op.cc
  49. conv_transpose_op.cc
  50. conv_transpose_op.h
  51. conv_transpose_op_cudnn.cc
  52. conv_transpose_op_gpu.cc
  53. conv_transpose_op_impl.h
  54. conv_transpose_op_mobile.cc
  55. conv_transpose_op_mobile.h
  56. conv_transpose_op_mobile_impl.h
  57. conv_transpose_unpool_op_base.h
  58. cos_op.cc
  59. cos_op.cu
  60. cosine_embedding_criterion_op.cc
  61. cosine_embedding_criterion_op.cu
  62. cosine_embedding_criterion_op.h
  63. counter_ops.cc
  64. counter_ops.h
  65. counter_ops_gpu.cc
  66. cross_entropy_op.cc
  67. cross_entropy_op.cu
  68. cross_entropy_op.h
  69. dataset_ops.cc
  70. dataset_ops.h
  71. distance_op.cc
  72. distance_op.cu
  73. distance_op.h
  74. dropout_op.cc
  75. dropout_op.cu
  76. dropout_op.h
  77. dropout_op_cudnn.cc
  78. elementwise_add_op.cc
  79. elementwise_div_op.cc
  80. elementwise_linear_op.cc
  81. elementwise_linear_op.cu
  82. elementwise_linear_op.h
  83. elementwise_logical_ops.cc
  84. elementwise_logical_ops.h
  85. elementwise_mul_op.cc
  86. elementwise_op.cc
  87. elementwise_op.cu
  88. elementwise_op.h
  89. elementwise_op_gpu_test.cc
  90. elementwise_op_schema.cc
  91. elementwise_op_test.cc
  92. elementwise_op_test.h
  93. elementwise_sub_op.cc
  94. elementwise_sum_op.cc
  95. elu_op.cc
  96. elu_op.cu
  97. elu_op.h
  98. exp_op.cc
  99. exp_op.cu
  100. extend_tensor_op.cc
  101. feed_blob_op.cc
  102. feed_blob_op.h
  103. filler_op.cc
  104. filler_op.cu
  105. filler_op.h
  106. find_duplicate_elements_op.cc
  107. find_duplicate_elements_op.h
  108. find_op.cc
  109. find_op.cu
  110. find_op.h
  111. free_op.cc
  112. free_op.h
  113. free_op_gpu.cc
  114. fully_connected_op.cc
  115. fully_connected_op.h
  116. fully_connected_op_gpu.cc
  117. fully_connected_op_gpu_test.cc
  118. fully_connected_op_test.cc
  119. given_tensor_fill_op.cc
  120. given_tensor_fill_op.cu
  121. given_tensor_fill_op.h
  122. gru_unit_op.cc
  123. gru_unit_op.h
  124. h_softmax_op.cc
  125. h_softmax_op.h
  126. half_float_ops.cc
  127. half_float_ops.cu
  128. half_float_ops.h
  129. im2col_op.cc
  130. im2col_op.h
  131. im2col_op_gpu.cc
  132. index_hash_ops.cc
  133. index_hash_ops.h
  134. index_ops.cc
  135. instance_norm_gradient_op.cc
  136. instance_norm_op.cc
  137. instance_norm_op.cu
  138. instance_norm_op.h
  139. last_n_window_collector.cc
  140. leaky_relu_op.cc
  141. leaky_relu_op.cu
  142. leaky_relu_op.h
  143. lengths_tile_op.cc
  144. lengths_tile_op.h
  145. load_save_op.cc
  146. load_save_op.h
  147. load_save_op_gpu.cc
  148. local_response_normalization_op.cc
  149. local_response_normalization_op.cu
  150. local_response_normalization_op.h
  151. local_response_normalization_op_cudnn.cc
  152. log_op.cc
  153. log_op.cu
  154. logit_op.cc
  155. loss_op.cc
  156. loss_op.cu
  157. loss_op.h
  158. lp_pool_op.cc
  159. lp_pool_op.cu
  160. lstm_unit_op.cc
  161. lstm_unit_op.h
  162. lstm_unit_op_gpu.cu
  163. map_ops.cc
  164. map_ops.h
  165. margin_ranking_criterion_op.cc
  166. margin_ranking_criterion_op.cu
  167. margin_ranking_criterion_op.h
  168. math_ops.cc
  169. math_ops.cu
  170. math_ops.h
  171. matmul_op.cc
  172. matmul_op.h
  173. matmul_op_gpu.cc
  174. max_pool_with_index.cu
  175. max_pool_with_index.h
  176. mem_query_op.cu
  177. multi_class_accuracy_op.cc
  178. multi_class_accuracy_op.cu
  179. multi_class_accuracy_op.h
  180. negative_op.cc
  181. negative_op.cu
  182. no_default_engine_op.h
  183. normalize_op.cc
  184. normalize_op.cu
  185. normalize_op.h
  186. one_hot_ops.cc
  187. one_hot_ops.h
  188. operator_fallback_gpu.h
  189. operator_fallback_gpu_test.cc
  190. order_switch_ops.cc
  191. order_switch_ops.cu
  192. order_switch_ops.h
  193. pack_rnn_sequence_op.cc
  194. pack_rnn_sequence_op.h
  195. pack_segments.cc
  196. pack_segments.h
  197. pack_segments_op_gpu.cc
  198. pad_op.cc
  199. pad_op.h
  200. pad_op_gpu.cu
  201. partition_ops.cc
  202. partition_ops.h
  203. perplexity_op.cc
  204. perplexity_op.cu
  205. perplexity_op.h
  206. piecewise_linear_transform_op.cc
  207. piecewise_linear_transform_op.cu
  208. piecewise_linear_transform_op.h
  209. pool_gradient_op.cc
  210. pool_op.cc
  211. pool_op.cu
  212. pool_op.h
  213. pool_op_cudnn.cc
  214. prefetch_op.h
  215. prelu_op.cc
  216. prelu_op.cu
  217. prelu_op.h
  218. rank_loss_op.cc
  219. rank_loss_op.h
  220. recurrent_network_blob_fetcher_op.cc
  221. recurrent_network_blob_fetcher_op.h
  222. recurrent_network_blob_fetcher_op_gpu.cc
  223. recurrent_network_executor.h
  224. recurrent_network_op.cc
  225. recurrent_network_op.h
  226. recurrent_network_op_gpu.cc
  227. recurrent_op_cudnn.cc
  228. recurrent_op_cudnn.h
  229. reducer_functors.h
  230. reduction_ops.cc
  231. reduction_ops.cu
  232. reduction_ops.h
  233. relu_op.cc
  234. relu_op.cu
  235. relu_op.h
  236. relu_op_cudnn.cc
  237. relu_op_fp16.cu
  238. remove_data_blocks_op.cc
  239. remove_data_blocks_op.h
  240. replace_nan_op.cc
  241. replace_nan_op.h
  242. reshape_op.cc
  243. reshape_op.h
  244. reshape_op_gpu.cc
  245. reshape_op_gpu_test.cc
  246. resize_op.cc
  247. resize_op.cu
  248. resize_op.h
  249. reverse_packed_segs_op.cc
  250. reverse_packed_segs_op.cu
  251. reverse_packed_segs_op.h
  252. roi_pool_op.cc
  253. roi_pool_op.cu
  254. roi_pool_op.h
  255. rowmul_op.cc
  256. rowmul_op.h
  257. scale_op.cc
  258. scale_op.h
  259. scale_op_gpu.cc
  260. segment_reduction_op.cc
  261. segment_reduction_op.h
  262. segment_reduction_op_gpu.cu
  263. sequence_ops.cc
  264. sequence_ops.h
  265. sigmoid_op.cc
  266. sigmoid_op.cu
  267. sin_op.cc
  268. sin_op.cu
  269. softmax_op.cc
  270. softmax_op.h
  271. softmax_op_cudnn.cc
  272. softmax_ops.cu
  273. softmax_shared.cc
  274. softmax_shared.h
  275. softmax_with_loss_op.cc
  276. softmax_with_loss_op.h
  277. softplus_op.cc
  278. softplus_op.cu
  279. softplus_op.h
  280. softsign_op.cc
  281. softsign_op.cu
  282. space_batch_op.cc
  283. space_batch_op.h
  284. space_batch_op_gpu.cu
  285. sparse_to_dense_mask_op.cc
  286. sparse_to_dense_op.cc
  287. sparse_to_dense_op.h
  288. spatial_batch_norm_gradient_op.cc
  289. spatial_batch_norm_op.cc
  290. spatial_batch_norm_op.h
  291. spatial_batch_norm_op_cudnn.cc
  292. spatial_softmax_with_loss_op.cc
  293. spatial_softmax_with_loss_op.h
  294. square_root_divide_op.cc
  295. square_root_divide_op.h
  296. stats_ops.cc
  297. stop_gradient.cc
  298. stop_gradient.h
  299. stop_gradient_gpu.cc
  300. string_ops.cc
  301. string_ops.h
  302. string_ops_test.cc
  303. summarize_op.cc
  304. summarize_op.cu
  305. summarize_op.h
  306. tanh_op.cc
  307. tanh_op.cu
  308. tensor_protos_db_input.cc
  309. tensor_protos_db_input.h
  310. tensor_protos_db_input_gpu.cc
  311. text_file_reader.cc
  312. text_file_reader_utils.cc
  313. text_file_reader_utils.h
  314. text_file_reader_utils_test.cc
  315. tile_op.cc
  316. tile_op.cu
  317. tile_op.h
  318. top_k.cc
  319. top_k.cu
  320. top_k.h
  321. top_k_heap_selection.cuh
  322. top_k_radix_selection.cuh
  323. transpose_op.cc
  324. transpose_op.cu
  325. transpose_op.h
  326. transpose_op_cudnn.cc
  327. tt_linear_op.cc
  328. tt_linear_op.h
  329. utility_ops.cc
  330. utility_ops.cu
  331. utility_ops.h
  332. utility_ops_gpu.cc
  333. utility_ops_gpu_test.cc
  334. utility_ops_test.cc
  335. workspace_ops.cc
  336. zero_gradient_op.cc
  337. zero_gradient_op.h
  338. zero_gradient_op_gpu.cc