Google Git
Sign in
android / platform / external / pytorch / 0d27d2686c / . / caffe2 / operators
tree: b8eac4d91f5dc2ffe1ddbfc2e8f0337a930408a3 [path history] [tgz]
  1. rnn/
  2. abs_op.cc
  3. abs_op.cu
  4. accumulate_op.cc
  5. accumulate_op.cu
  6. accumulate_op.h
  7. accuracy_op.cc
  8. accuracy_op.cu
  9. accuracy_op.h
  10. acos_op.cc
  11. acos_op.cu
  12. apmeter_op.cc
  13. apmeter_op.h
  14. arg_max_op.cc
  15. arg_max_op.h
  16. arg_ops.cc
  17. arg_ops.cu
  18. arg_ops.h
  19. arg_ops_eigen.h
  20. asin_op.cc
  21. asin_op.cu
  22. assert_op.cc
  23. assert_op.cu
  24. assert_op.h
  25. atan_op.cc
  26. atan_op.cu
  27. atomic_ops.cc
  28. batch_box_cox_op.cc
  29. batch_box_cox_op.h
  30. batch_gather_ops.cc
  31. batch_gather_ops.cu
  32. batch_gather_ops.h
  33. batch_matmul_op.cc
  34. batch_matmul_op.cu
  35. batch_matmul_op.h
  36. batch_matmul_op_gpu_test.cc
  37. batch_matmul_op_test.cc
  38. batch_sparse_to_dense_op.cc
  39. batch_sparse_to_dense_op.h
  40. bbox_transform_op.cc
  41. bbox_transform_op.h
  42. boolean_mask_ops.cc
  43. boolean_mask_ops.cu
  44. boolean_mask_ops.h
  45. boolean_unmask_ops.cc
  46. boolean_unmask_ops.cu
  47. boolean_unmask_ops.h
  48. boolean_unmask_ops_test.cc
  49. box_with_nms_limit_op.cc
  50. box_with_nms_limit_op.h
  51. cast_op.cc
  52. cast_op.cu
  53. cast_op.h
  54. ceil_op.cc
  55. ceil_op.cu
  56. ceil_op.h
  57. channel_backprop_stats_op.cc
  58. channel_backprop_stats_op.cu
  59. channel_backprop_stats_op.h
  60. channel_shuffle_op.cc
  61. channel_shuffle_op.h
  62. channel_shuffle_op_gpu.cu
  63. channel_stats_op.cc
  64. channel_stats_op.cu
  65. channel_stats_op.h
  66. clip_op.cc
  67. clip_op.cu
  68. clip_op.h
  69. CMakeLists.txt
  70. collect_and_distribute_fpn_rpn_proposals_op.cc
  71. collect_and_distribute_fpn_rpn_proposals_op.h
  72. communicator_op.cc
  73. communicator_op_gpu.cc
  74. concat_split_op.cc
  75. concat_split_op.h
  76. concat_split_op_gpu.cc
  77. conditional_op.cc
  78. conditional_op.h
  79. conv_gradient_op.cc
  80. conv_op.cc
  81. conv_op.h
  82. conv_op_cache_cudnn.cc
  83. conv_op_cache_cudnn.h
  84. conv_op_cache_cudnn_test.cc
  85. conv_op_cudnn.cc
  86. conv_op_eigen.cc
  87. conv_op_gpu.cc
  88. conv_op_impl.h
  89. conv_op_shared.cc
  90. conv_op_shared.h
  91. conv_op_shared_gpu.cc
  92. conv_pool_op_base.h
  93. conv_transpose_gradient_op.cc
  94. conv_transpose_op.cc
  95. conv_transpose_op.h
  96. conv_transpose_op_cudnn.cc
  97. conv_transpose_op_gpu.cc
  98. conv_transpose_op_impl.h
  99. conv_transpose_op_mobile.cc
  100. conv_transpose_op_mobile.h
  101. conv_transpose_op_mobile_impl.h
  102. conv_transpose_op_mobile_test.cc
  103. conv_transpose_unpool_op_base.h
  104. cos_op.cc
  105. cos_op.cu
  106. cosine_embedding_criterion_op.cc
  107. cosine_embedding_criterion_op.cu
  108. cosine_embedding_criterion_op.h
  109. counter_ops.cc
  110. counter_ops.h
  111. counter_ops_gpu.cc
  112. create_scope_op.cc
  113. create_scope_op.h
  114. cross_entropy_op.cc
  115. cross_entropy_op.cu
  116. cross_entropy_op.h
  117. dataset_ops.cc
  118. dataset_ops.h
  119. deform_conv_gradient_op.cc
  120. deform_conv_op.cc
  121. deform_conv_op.cu
  122. deform_conv_op.h
  123. deform_conv_op_impl.h
  124. depthwise_3x3_conv_op.cu
  125. distance_op.cc
  126. distance_op.cu
  127. distance_op.h
  128. do_op.cc
  129. do_op.h
  130. do_op_gpu.cc
  131. dropout_op.cc
  132. dropout_op.cu
  133. dropout_op.h
  134. dropout_op_cudnn.cc
  135. elementwise_add_op.cc
  136. elementwise_div_op.cc
  137. elementwise_linear_op.cc
  138. elementwise_linear_op.cu
  139. elementwise_linear_op.h
  140. elementwise_logical_ops.cc
  141. elementwise_logical_ops.h
  142. elementwise_mul_op.cc
  143. elementwise_op.cc
  144. elementwise_op.cu
  145. elementwise_op.h
  146. elementwise_op_gpu_test.cc
  147. elementwise_op_schema.cc
  148. elementwise_op_test.cc
  149. elementwise_op_test.h
  150. elementwise_sub_op.cc
  151. elementwise_sum_op.cc
  152. elu_op.cc
  153. elu_op.cu
  154. elu_op.h
  155. enforce_finite_op.cc
  156. enforce_finite_op.cu
  157. enforce_finite_op.h
  158. ensure_cpu_output_op.cc
  159. ensure_cpu_output_op.cu
  160. ensure_cpu_output_op.h
  161. exp_op.cc
  162. exp_op.cu
  163. expand_squeeze_dims_op.cc
  164. expand_squeeze_dims_op.h
  165. expand_squeeze_dims_op_gpu.cc
  166. extend_tensor_op.cc
  167. fc_inference.cc
  168. fc_inference.h
  169. feature_maps_ops.cc
  170. feature_maps_ops.h
  171. feed_blob_op.cc
  172. feed_blob_op.h
  173. filler_op.cc
  174. filler_op.cu
  175. filler_op.h
  176. filler_op_gpu.cc
  177. find_duplicate_elements_op.cc
  178. find_duplicate_elements_op.h
  179. find_op.cc
  180. find_op.cu
  181. find_op.h
  182. flatten_op.cc
  183. flatten_op.h
  184. flexible_top_k.cc
  185. flexible_top_k.h
  186. floor_op.cc
  187. floor_op.cu
  188. floor_op.h
  189. free_op.cc
  190. free_op.h
  191. free_op_gpu.cc
  192. fully_connected_op.cc
  193. fully_connected_op.h
  194. fully_connected_op_gpu.cc
  195. fused_rowwise_8bit_conversion_ops.cc
  196. fused_rowwise_8bit_conversion_ops.h
  197. gather_fused_8bit_rowwise_op.cc
  198. gather_fused_8bit_rowwise_op.h
  199. gather_ranges_to_dense_op.cc
  200. gather_ranges_to_dense_op.h
  201. generate_proposals_op.cc
  202. generate_proposals_op.h
  203. generate_proposals_op_test.cc
  204. generate_proposals_op_util_boxes.h
  205. generate_proposals_op_util_boxes_test.cc
  206. generate_proposals_op_util_nms.h
  207. generate_proposals_op_util_nms_test.cc
  208. given_tensor_fill_op.cc
  209. given_tensor_fill_op.cu
  210. given_tensor_fill_op.h
  211. glu_op.cc
  212. glu_op.cu
  213. glu_op.h
  214. group_norm_op.cc
  215. group_norm_op.cu
  216. group_norm_op.h
  217. gru_unit_op.cc
  218. gru_unit_op.h
  219. gru_unit_op_gpu.cu
  220. h_softmax_op.cc
  221. h_softmax_op.h
  222. half_float_ops.cc
  223. half_float_ops.cu
  224. half_float_ops.h
  225. if_op.cc
  226. if_op.h
  227. if_op_gpu.cc
  228. im2col_op.cc
  229. im2col_op.h
  230. im2col_op_gpu.cc
  231. index_hash_ops.cc
  232. index_hash_ops.h
  233. index_ops.cc
  234. instance_norm_gradient_op.cc
  235. instance_norm_op.cc
  236. instance_norm_op.cu
  237. instance_norm_op.h
  238. integral_image_op.cc
  239. integral_image_op.cu
  240. integral_image_op.h
  241. jsd_op.cc
  242. jsd_op.h
  243. key_split_ops.cc
  244. key_split_ops.h
  245. last_n_window_collector.cc
  246. layer_norm_op.cc
  247. layer_norm_op.cu
  248. layer_norm_op.h
  249. leaky_relu_op.cc
  250. leaky_relu_op.cu
  251. leaky_relu_op.h
  252. lengths_pad_op.cc
  253. lengths_pad_op.cu
  254. lengths_pad_op.h
  255. lengths_reducer_fused_8bit_rowwise_ops.cc
  256. lengths_reducer_fused_8bit_rowwise_ops.h
  257. lengths_reducer_ops.cc
  258. lengths_reducer_ops.h
  259. lengths_reducer_rowwise_8bit_ops.cc
  260. lengths_reducer_rowwise_8bit_ops.h
  261. lengths_tile_op.cc
  262. lengths_tile_op.h
  263. lengths_tile_op_gpu.cc
  264. lengths_top_k_op.cc
  265. lengths_top_k_op.h
  266. listwise_l2r_op.cc
  267. listwise_l2r_op.h
  268. load_save_op.cc
  269. load_save_op.h
  270. load_save_op_gpu.cc
  271. local_response_normalization_op.cc
  272. local_response_normalization_op.cu
  273. local_response_normalization_op.h
  274. local_response_normalization_op_cudnn.cc
  275. locally_connected_op.cc
  276. locally_connected_op.h
  277. locally_connected_op_gpu.cc
  278. locally_connected_op_impl.h
  279. locally_connected_op_util.cc
  280. locally_connected_op_util.h
  281. log_op.cc
  282. log_op.cu
  283. logit_op.cc
  284. logit_op.cu
  285. logit_op.h
  286. loss_op.cc
  287. loss_op.cu
  288. loss_op.h
  289. lp_pool_op.cc
  290. lp_pool_op.cu
  291. lpnorm_op.cc
  292. lpnorm_op.h
  293. lstm_unit_op.cc
  294. lstm_unit_op.h
  295. lstm_unit_op_gpu.cu
  296. map_ops.cc
  297. map_ops.h
  298. margin_ranking_criterion_op.cc
  299. margin_ranking_criterion_op.cu
  300. margin_ranking_criterion_op.h
  301. math_ops.cc
  302. math_ops.cu
  303. math_ops.h
  304. matmul_op.cc
  305. matmul_op.h
  306. matmul_op_gpu.cc
  307. max_pool_with_index.cu
  308. max_pool_with_index.h
  309. mean_op.cc
  310. mean_op.cu
  311. mean_op.h
  312. mem_query_op.cu
  313. merge_id_lists_op.cc
  314. merge_id_lists_op.h
  315. minmax_gradient_ops.cc
  316. minmax_ops.cc
  317. minmax_ops.h
  318. mod_op.cc
  319. mod_op.h
  320. moments_op.cc
  321. moments_op.cu
  322. moments_op.h
  323. multi_class_accuracy_op.cc
  324. multi_class_accuracy_op.cu
  325. multi_class_accuracy_op.h
  326. negate_gradient_op.cc
  327. negate_gradient_op.h
  328. negate_gradient_op_gpu.cc
  329. negative_op.cc
  330. negative_op.cu
  331. ngram_ops.cc
  332. ngram_ops.h
  333. no_default_engine_op.h
  334. norm_planar_yuv_op.cc
  335. normalize_l1_op.cc
  336. normalize_l1_op.h
  337. normalize_op.cc
  338. normalize_op.h
  339. normalize_ops.cu
  340. numpy_tile_op.cc
  341. numpy_tile_op.h
  342. one_hot_ops.cc
  343. one_hot_ops.cu
  344. one_hot_ops.h
  345. onnx_while_op.cc
  346. onnx_while_op.h
  347. op_utils_cudnn.h
  348. operator_fallback_gpu.h
  349. operator_fallback_gpu_test.cc
  350. order_switch_ops.cc
  351. order_switch_ops.cu
  352. order_switch_ops.h
  353. pack_rnn_sequence_op.cc
  354. pack_rnn_sequence_op.h
  355. pack_segments.cc
  356. pack_segments.cu
  357. pack_segments.h
  358. pad_op.cc
  359. pad_op.h
  360. pad_op_gpu.cu
  361. partition_ops.cc
  362. partition_ops.h
  363. percentile_op.cc
  364. percentile_op.h
  365. perplexity_op.cc
  366. perplexity_op.cu
  367. perplexity_op.h
  368. piecewise_linear_transform_op.cc
  369. piecewise_linear_transform_op.cu
  370. piecewise_linear_transform_op.h
  371. pool_gradient_op.cc
  372. pool_op.cc
  373. pool_op.cu
  374. pool_op.h
  375. pool_op_cudnn.cu
  376. pow_op.cc
  377. pow_op.cu
  378. pow_op.h
  379. prefetch_op.h
  380. prelu_op.cc
  381. prelu_op.cu
  382. prelu_op.h
  383. prepend_dim_op.cc
  384. prepend_dim_op.h
  385. prepend_dim_op_gpu.cc
  386. quant_decode_op.cc
  387. quant_decode_op.h
  388. rank_loss_op.cc
  389. rank_loss_op.h
  390. reduce_ops.cc
  391. reduce_ops.cu
  392. reduce_ops.h
  393. reducer_functors.h
  394. reduction_front_back_ops.cc
  395. reduction_front_back_ops.cu
  396. reduction_front_back_ops.h
  397. reduction_ops.cc
  398. reduction_ops.cu
  399. reduction_ops.h
  400. relu_n_op.cc
  401. relu_n_op.cu
  402. relu_n_op.h
  403. relu_op.cc
  404. relu_op.cu
  405. relu_op.h
  406. relu_op_cudnn.cc
  407. relu_op_fp16.cu
  408. remove_data_blocks_op.cc
  409. remove_data_blocks_op.h
  410. replace_nan_op.cc
  411. replace_nan_op.cu
  412. replace_nan_op.h
  413. reservoir_sampling.cc
  414. reshape_op.cc
  415. reshape_op.h
  416. reshape_op_gpu.cc
  417. reshape_op_gpu_test.cc
  418. resize_op.cc
  419. resize_op.cu
  420. resize_op.h
  421. reverse_packed_segs_op.cc
  422. reverse_packed_segs_op.cu
  423. reverse_packed_segs_op.h
  424. rmac_regions_op.cc
  425. rmac_regions_op.cu
  426. rmac_regions_op.h
  427. roi_align_gradient_op.cc
  428. roi_align_gradient_op.cu
  429. roi_align_gradient_op.h
  430. roi_align_op.cc
  431. roi_align_op.cu
  432. roi_align_op.h
  433. roi_align_op_gpu_test.cc
  434. roi_pool_op.cc
  435. roi_pool_op.cu
  436. roi_pool_op.h
  437. rowmul_op.cc
  438. rowmul_op.h
  439. rsqrt_op.cc
  440. rsqrt_op.cu
  441. rsqrt_op.h
  442. scale_op.cc
  443. scale_op.h
  444. scale_op_gpu.cc
  445. segment_reduction_op.cc
  446. segment_reduction_op.h
  447. segment_reduction_op_gpu.cu
  448. selu_op.cc
  449. selu_op.cu
  450. selu_op.h
  451. sequence_ops.cc
  452. sequence_ops.cu
  453. sequence_ops.h
  454. shape_op.cc
  455. shape_op.h
  456. shape_op_gpu.cc
  457. sigmoid_op.cc
  458. sigmoid_op.cu
  459. sin_op.cc
  460. sin_op.cu
  461. sinusoid_position_encoding_op.cc
  462. sinusoid_position_encoding_op.h
  463. slice_op.cc
  464. slice_op.cu
  465. slice_op.h
  466. softmax_op.cc
  467. softmax_op.h
  468. softmax_op_cudnn.cc
  469. softmax_ops.cu
  470. softmax_shared.cc
  471. softmax_shared.h
  472. softmax_with_loss_op.cc
  473. softmax_with_loss_op.h
  474. softplus_op.cc
  475. softplus_op.cu
  476. softplus_op.h
  477. softsign_op.cc
  478. softsign_op.cu
  479. space_batch_op.cc
  480. space_batch_op.h
  481. space_batch_op_gpu.cu
  482. sparse_normalize_op.cc
  483. sparse_normalize_op.h
  484. sparse_to_dense_mask_op.cc
  485. sparse_to_dense_mask_op.h
  486. sparse_to_dense_op.cc
  487. sparse_to_dense_op.cu
  488. sparse_to_dense_op.h
  489. spatial_batch_norm_gradient_op.cc
  490. spatial_batch_norm_op.cc
  491. spatial_batch_norm_op.h
  492. spatial_batch_norm_op_cudnn.cc
  493. spatial_softmax_with_loss_op.cc
  494. spatial_softmax_with_loss_op.h
  495. sqrt_op.cc
  496. sqrt_op.cu
  497. square_root_divide_op.cc
  498. square_root_divide_op.h
  499. stats_ops.cc
  500. stop_gradient.cc
  501. stop_gradient.h
  502. stop_gradient_gpu.cc
  503. string_ops.cc
  504. string_ops.h
  505. string_ops_test.cc
  506. stump_func_op.cc
  507. stump_func_op.cu
  508. stump_func_op.h
  509. stylizer_ops.cc
  510. summarize_op.cc
  511. summarize_op.cu
  512. summarize_op.h
  513. swish_op.cc
  514. swish_op.cu
  515. swish_op.h
  516. tan_op.cc
  517. tan_op.cu
  518. tanh_op.cc
  519. tanh_op.cu
  520. tensor_protos_db_input.cc
  521. tensor_protos_db_input.h
  522. tensor_protos_db_input_gpu.cc
  523. text_file_reader.cc
  524. text_file_reader_utils.cc
  525. text_file_reader_utils.h
  526. text_file_reader_utils_test.cc
  527. thresholded_relu_op.cc
  528. thresholded_relu_op.cu
  529. thresholded_relu_op.h
  530. tile_op.cc
  531. tile_op.cu
  532. tile_op.h
  533. top_k.cc
  534. top_k.cu
  535. top_k.h
  536. top_k_heap_selection.cuh
  537. top_k_radix_selection.cuh
  538. transpose_op.cc
  539. transpose_op.cu
  540. transpose_op.h
  541. transpose_op_cudnn.cc
  542. tt_linear_op.cc
  543. tt_linear_op.h
  544. unique_ops.cc
  545. unique_ops.cu
  546. unique_ops.h
  547. utility_ops.cc
  548. utility_ops.cu
  549. utility_ops.h
  550. utility_ops_gpu.cc
  551. utility_ops_gpu_test.cc
  552. utility_ops_test.cc
  553. variable_length_sequence_padding.cc
  554. variable_length_sequence_padding.h
  555. weighted_multi_sampling_op.cc
  556. weighted_multi_sampling_op.h
  557. weighted_sample_op.cc
  558. weighted_sample_op.cu
  559. weighted_sample_op.h
  560. while_op.cc
  561. while_op.h
  562. while_op_gpu.cc
  563. workspace_ops.cc
  564. zero_gradient_op.cc
  565. zero_gradient_op.h
  566. zero_gradient_op_gpu.cc