Google Git
Sign in
android / platform / external / pytorch / 2af506cb6c / . / caffe2 / operators
tree: c0a1ff59e452351b9dca43ba0e161976aa6fcae9 [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. assert_op.cc
  12. assert_op.cu
  13. assert_op.h
  14. atomic_ops.cc
  15. batch_box_cox_op.cc
  16. batch_box_cox_op.h
  17. batch_gather_ops.cc
  18. batch_gather_ops.cu
  19. batch_gather_ops.h
  20. batch_matmul_op.cc
  21. batch_matmul_op.cu
  22. batch_matmul_op.h
  23. batch_sparse_to_dense_op.cc
  24. batch_sparse_to_dense_op.h
  25. boolean_mask_ops.cc
  26. boolean_mask_ops.cu
  27. boolean_mask_ops.h
  28. boolean_unmask_ops.cc
  29. boolean_unmask_ops.cu
  30. boolean_unmask_ops.h
  31. boolean_unmask_ops_test.cc
  32. cast_op.cc
  33. cast_op.cu
  34. cast_op.h
  35. channel_shuffle_op.cc
  36. channel_shuffle_op.h
  37. channel_shuffle_op_gpu.cu
  38. clip_op.cc
  39. clip_op.cu
  40. clip_op.h
  41. CMakeLists.txt
  42. communicator_op.cc
  43. communicator_op_gpu.cc
  44. concat_split_op.cc
  45. concat_split_op.h
  46. concat_split_op_gpu.cc
  47. conditional_op.cc
  48. conditional_op.h
  49. conv_gradient_op.cc
  50. conv_op.cc
  51. conv_op.h
  52. conv_op_cache_cudnn.cc
  53. conv_op_cache_cudnn.h
  54. conv_op_cache_cudnn_test.cc
  55. conv_op_cudnn.cc
  56. conv_op_eigen.cc
  57. conv_op_gpu.cc
  58. conv_op_impl.h
  59. conv_op_shared.cc
  60. conv_op_shared.h
  61. conv_op_shared_gpu.cc
  62. conv_pool_op_base.h
  63. conv_transpose_gradient_op.cc
  64. conv_transpose_op.cc
  65. conv_transpose_op.h
  66. conv_transpose_op_cudnn.cc
  67. conv_transpose_op_gpu.cc
  68. conv_transpose_op_impl.h
  69. conv_transpose_op_mobile.cc
  70. conv_transpose_op_mobile.h
  71. conv_transpose_op_mobile_impl.h
  72. conv_transpose_op_mobile_test.cc
  73. conv_transpose_unpool_op_base.h
  74. cos_op.cc
  75. cos_op.cu
  76. cosine_embedding_criterion_op.cc
  77. cosine_embedding_criterion_op.cu
  78. cosine_embedding_criterion_op.h
  79. counter_ops.cc
  80. counter_ops.h
  81. counter_ops_gpu.cc
  82. create_scope_op.cc
  83. create_scope_op.h
  84. cross_entropy_op.cc
  85. cross_entropy_op.cu
  86. cross_entropy_op.h
  87. dataset_ops.cc
  88. dataset_ops.h
  89. deform_conv_gradient_op.cc
  90. deform_conv_op.cc
  91. deform_conv_op.cu
  92. deform_conv_op.h
  93. deform_conv_op_impl.h
  94. distance_op.cc
  95. distance_op.cu
  96. distance_op.h
  97. do_op.cc
  98. do_op.h
  99. do_op_gpu.cc
  100. dropout_op.cc
  101. dropout_op.cu
  102. dropout_op.h
  103. dropout_op_cudnn.cc
  104. elementwise_add_op.cc
  105. elementwise_div_op.cc
  106. elementwise_linear_op.cc
  107. elementwise_linear_op.cu
  108. elementwise_linear_op.h
  109. elementwise_logical_ops.cc
  110. elementwise_logical_ops.h
  111. elementwise_mul_op.cc
  112. elementwise_op.cc
  113. elementwise_op.cu
  114. elementwise_op.h
  115. elementwise_op_gpu_test.cc
  116. elementwise_op_schema.cc
  117. elementwise_op_test.cc
  118. elementwise_op_test.h
  119. elementwise_sub_op.cc
  120. elementwise_sum_op.cc
  121. elu_op.cc
  122. elu_op.cu
  123. elu_op.h
  124. exp_op.cc
  125. exp_op.cu
  126. expand_squeeze_dims_op.cc
  127. expand_squeeze_dims_op.h
  128. expand_squeeze_dims_op_gpu.cc
  129. extend_tensor_op.cc
  130. feed_blob_op.cc
  131. feed_blob_op.h
  132. filler_op.cc
  133. filler_op.cu
  134. filler_op.h
  135. filler_op_gpu.cc
  136. find_duplicate_elements_op.cc
  137. find_duplicate_elements_op.h
  138. find_op.cc
  139. find_op.cu
  140. find_op.h
  141. flatten_op.cc
  142. flatten_op.h
  143. flexible_top_k.cc
  144. flexible_top_k.h
  145. free_op.cc
  146. free_op.h
  147. free_op_gpu.cc
  148. fully_connected_op.cc
  149. fully_connected_op.h
  150. fully_connected_op_gpu.cc
  151. fully_connected_op_gpu_test.cc
  152. fully_connected_op_test.cc
  153. gather_ranges_to_dense_op.cc
  154. gather_ranges_to_dense_op.h
  155. generate_proposals_op.h
  156. generate_proposals_op_util_boxes.h
  157. given_tensor_fill_op.cc
  158. given_tensor_fill_op.cu
  159. given_tensor_fill_op.h
  160. glu_op.cc
  161. glu_op.cu
  162. glu_op.h
  163. gru_unit_op.cc
  164. gru_unit_op.h
  165. gru_unit_op_gpu.cu
  166. h_softmax_op.cc
  167. h_softmax_op.h
  168. half_float_ops.cc
  169. half_float_ops.cu
  170. half_float_ops.h
  171. if_op.cc
  172. if_op.h
  173. if_op_gpu.cc
  174. im2col_op.cc
  175. im2col_op.h
  176. im2col_op_gpu.cc
  177. index_hash_ops.cc
  178. index_hash_ops.h
  179. index_ops.cc
  180. instance_norm_gradient_op.cc
  181. instance_norm_op.cc
  182. instance_norm_op.cu
  183. instance_norm_op.h
  184. last_n_window_collector.cc
  185. layer_norm_op.cc
  186. layer_norm_op.cu
  187. layer_norm_op.h
  188. leaky_relu_op.cc
  189. leaky_relu_op.cu
  190. leaky_relu_op.h
  191. lengths_reducer_ops.cc
  192. lengths_reducer_ops.h
  193. lengths_reducer_rowwise_8bit_ops.cc
  194. lengths_reducer_rowwise_8bit_ops.h
  195. lengths_tile_op.cc
  196. lengths_tile_op.h
  197. lengths_tile_op_gpu.cc
  198. lengths_top_k_op.cc
  199. lengths_top_k_op.h
  200. listwise_l2r_op.cc
  201. listwise_l2r_op.h
  202. load_save_op.cc
  203. load_save_op.h
  204. load_save_op_gpu.cc
  205. local_response_normalization_op.cc
  206. local_response_normalization_op.cu
  207. local_response_normalization_op.h
  208. local_response_normalization_op_cudnn.cc
  209. log_op.cc
  210. log_op.cu
  211. logit_op.cc
  212. logit_op.cu
  213. logit_op.h
  214. loss_op.cc
  215. loss_op.cu
  216. loss_op.h
  217. lp_pool_op.cc
  218. lp_pool_op.cu
  219. lpnorm_op.cc
  220. lpnorm_op.h
  221. lstm_unit_op.cc
  222. lstm_unit_op.h
  223. lstm_unit_op_gpu.cu
  224. map_ops.cc
  225. map_ops.h
  226. margin_ranking_criterion_op.cc
  227. margin_ranking_criterion_op.cu
  228. margin_ranking_criterion_op.h
  229. math_ops.cc
  230. math_ops.cu
  231. math_ops.h
  232. matmul_op.cc
  233. matmul_op.h
  234. matmul_op_gpu.cc
  235. max_pool_with_index.cu
  236. max_pool_with_index.h
  237. mem_query_op.cu
  238. merge_id_lists_op.cc
  239. merge_id_lists_op.h
  240. minmax_gradient_ops.cc
  241. minmax_ops.cc
  242. minmax_ops.h
  243. mod_op.cc
  244. mod_op.h
  245. multi_class_accuracy_op.cc
  246. multi_class_accuracy_op.cu
  247. multi_class_accuracy_op.h
  248. negate_gradient_op.cc
  249. negate_gradient_op.h
  250. negate_gradient_op_gpu.cc
  251. negative_op.cc
  252. negative_op.cu
  253. no_default_engine_op.h
  254. norm_planar_yuv_op.cc
  255. normalize_l1_op.cc
  256. normalize_l1_op.h
  257. normalize_op.cc
  258. normalize_op.h
  259. normalize_ops.cu
  260. one_hot_ops.cc
  261. one_hot_ops.cu
  262. one_hot_ops.h
  263. operator_fallback_gpu.h
  264. operator_fallback_gpu_test.cc
  265. order_switch_ops.cc
  266. order_switch_ops.cu
  267. order_switch_ops.h
  268. pack_rnn_sequence_op.cc
  269. pack_rnn_sequence_op.h
  270. pack_segments.cc
  271. pack_segments.cu
  272. pack_segments.h
  273. pad_op.cc
  274. pad_op.h
  275. pad_op_gpu.cu
  276. partition_ops.cc
  277. partition_ops.h
  278. perplexity_op.cc
  279. perplexity_op.cu
  280. perplexity_op.h
  281. piecewise_linear_transform_op.cc
  282. piecewise_linear_transform_op.cu
  283. piecewise_linear_transform_op.h
  284. pool_gradient_op.cc
  285. pool_op.cc
  286. pool_op.cu
  287. pool_op.h
  288. pool_op_cudnn.cu
  289. prefetch_op.h
  290. prelu_op.cc
  291. prelu_op.cu
  292. prelu_op.h
  293. prepend_dim_op.cc
  294. prepend_dim_op.h
  295. prepend_dim_op_gpu.cc
  296. quant_decode_op.cc
  297. quant_decode_op.h
  298. rank_loss_op.cc
  299. rank_loss_op.h
  300. recurrent_network_blob_fetcher_op.cc
  301. recurrent_network_blob_fetcher_op.h
  302. recurrent_network_blob_fetcher_op_gpu.cc
  303. recurrent_network_executor.cc
  304. recurrent_network_executor.h
  305. recurrent_network_executor_gpu.cc
  306. recurrent_network_executor_gpu.h
  307. recurrent_network_executor_incl.h
  308. recurrent_network_op.cc
  309. recurrent_network_op.h
  310. recurrent_network_op_gpu.cu
  311. recurrent_op_cudnn.cc
  312. recurrent_op_cudnn.h
  313. reducer_functors.h
  314. reduction_front_back_ops.cc
  315. reduction_front_back_ops.cu
  316. reduction_front_back_ops.h
  317. reduction_ops.cc
  318. reduction_ops.cu
  319. reduction_ops.h
  320. relu_op.cc
  321. relu_op.cu
  322. relu_op.h
  323. relu_op_cudnn.cc
  324. relu_op_fp16.cu
  325. remove_data_blocks_op.cc
  326. remove_data_blocks_op.h
  327. replace_nan_op.cc
  328. replace_nan_op.cu
  329. replace_nan_op.h
  330. reservoir_sampling.cc
  331. reshape_op.cc
  332. reshape_op.h
  333. reshape_op_gpu.cc
  334. reshape_op_gpu_test.cc
  335. resize_op.cc
  336. resize_op.cu
  337. resize_op.h
  338. reverse_packed_segs_op.cc
  339. reverse_packed_segs_op.cu
  340. reverse_packed_segs_op.h
  341. rmac_regions_op.cc
  342. rmac_regions_op.cu
  343. rmac_regions_op.h
  344. roi_pool_op.cc
  345. roi_pool_op.cu
  346. roi_pool_op.h
  347. rowmul_op.cc
  348. rowmul_op.h
  349. scale_op.cc
  350. scale_op.h
  351. scale_op_gpu.cc
  352. segment_reduction_op.cc
  353. segment_reduction_op.h
  354. segment_reduction_op_gpu.cu
  355. selu_op.cc
  356. selu_op.cu
  357. selu_op.h
  358. sequence_ops.cc
  359. sequence_ops.cu
  360. sequence_ops.h
  361. shape_op.cc
  362. shape_op.h
  363. shape_op_gpu.cc
  364. sigmoid_op.cc
  365. sigmoid_op.cu
  366. sin_op.cc
  367. sin_op.cu
  368. sinusoid_position_encoding_op.cc
  369. sinusoid_position_encoding_op.h
  370. slice_op.cc
  371. slice_op.cu
  372. slice_op.h
  373. softmax_op.cc
  374. softmax_op.h
  375. softmax_op_cudnn.cc
  376. softmax_ops.cu
  377. softmax_shared.cc
  378. softmax_shared.h
  379. softmax_with_loss_op.cc
  380. softmax_with_loss_op.h
  381. softplus_op.cc
  382. softplus_op.cu
  383. softplus_op.h
  384. softsign_op.cc
  385. softsign_op.cu
  386. space_batch_op.cc
  387. space_batch_op.h
  388. space_batch_op_gpu.cu
  389. sparse_to_dense_mask_op.cc
  390. sparse_to_dense_mask_op.h
  391. sparse_to_dense_op.cc
  392. sparse_to_dense_op.cu
  393. sparse_to_dense_op.h
  394. spatial_batch_norm_gradient_op.cc
  395. spatial_batch_norm_op.cc
  396. spatial_batch_norm_op.h
  397. spatial_batch_norm_op_cudnn.cc
  398. spatial_softmax_with_loss_op.cc
  399. spatial_softmax_with_loss_op.h
  400. square_root_divide_op.cc
  401. square_root_divide_op.h
  402. stats_ops.cc
  403. stop_gradient.cc
  404. stop_gradient.h
  405. stop_gradient_gpu.cc
  406. string_ops.cc
  407. string_ops.h
  408. string_ops_test.cc
  409. stylizer_ops.cc
  410. summarize_op.cc
  411. summarize_op.cu
  412. summarize_op.h
  413. swish_op.cc
  414. swish_op.h
  415. tanh_op.cc
  416. tanh_op.cu
  417. tensor_protos_db_input.cc
  418. tensor_protos_db_input.h
  419. tensor_protos_db_input_gpu.cc
  420. text_file_reader.cc
  421. text_file_reader_utils.cc
  422. text_file_reader_utils.h
  423. text_file_reader_utils_test.cc
  424. tile_op.cc
  425. tile_op.cu
  426. tile_op.h
  427. top_k.cc
  428. top_k.cu
  429. top_k.h
  430. top_k_heap_selection.cuh
  431. top_k_radix_selection.cuh
  432. transpose_op.cc
  433. transpose_op.cu
  434. transpose_op.h
  435. transpose_op_cudnn.cc
  436. transpose_op_gpu.h
  437. tt_linear_op.cc
  438. tt_linear_op.h
  439. utility_ops.cc
  440. utility_ops.cu
  441. utility_ops.h
  442. utility_ops_gpu.cc
  443. utility_ops_gpu_test.cc
  444. utility_ops_test.cc
  445. weighted_sample_op.cc
  446. weighted_sample_op.cu
  447. weighted_sample_op.h
  448. while_op.cc
  449. while_op.h
  450. while_op_gpu.cc
  451. workspace_ops.cc
  452. zero_gradient_op.cc
  453. zero_gradient_op.h
  454. zero_gradient_op_gpu.cc