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