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