Google Git
Sign in
android / platform / external / tensorflow / 6341b8975b7660244e1ca3003bfce5371f1fd167 / . / tensorflow / core / kernels
tree: a488de9e57f4f3394e7dd60bf8852035686b5e26 [path history] [tgz]
  1. batching_util/
  2. boosted_trees/
  3. data/
  4. fuzzing/
  5. hexagon/
  6. neon/
  7. rnn/
  8. spectrogram_test_data/
  9. tensor_forest/
  10. adjust_contrast_op.cc
  11. adjust_contrast_op.h
  12. adjust_contrast_op_benchmark_test.cc
  13. adjust_contrast_op_gpu.cu.cc
  14. adjust_contrast_op_test.cc
  15. adjust_hsv_gpu.cu.h
  16. adjust_hue_op.cc
  17. adjust_hue_op.h
  18. adjust_hue_op_gpu.cu.cc
  19. adjust_saturation_op.cc
  20. adjust_saturation_op.h
  21. adjust_saturation_op_gpu.cu.cc
  22. aggregate_ops.cc
  23. aggregate_ops.h
  24. aggregate_ops_cpu.h
  25. aggregate_ops_gpu.cu.cc
  26. argmax_op.cc
  27. argmax_op.h
  28. argmax_op_gpu.cu.cc
  29. as_string_op.cc
  30. assign_op.h
  31. attention_ops.cc
  32. avgpooling_op.cc
  33. avgpooling_op.h
  34. avgpooling_op_gpu.cu.cc
  35. barrier_ops.cc
  36. base64_ops.cc
  37. basic_ops_benchmark_test.cc
  38. batch_kernels.cc
  39. batch_matmul_op_complex.cc
  40. batch_matmul_op_impl.h
  41. batch_matmul_op_real.cc
  42. batch_matmul_op_test.cc
  43. batch_norm_op.cc
  44. batch_norm_op.h
  45. batch_norm_op_gpu.cu.cc
  46. batch_norm_op_test.cc
  47. batch_util.h
  48. batchtospace_op.cc
  49. bcast_ops.cc
  50. betainc_op.cc
  51. betainc_op.h
  52. betainc_op_gpu.cu.cc
  53. bias_op.cc
  54. bias_op.h
  55. bias_op_gpu.cu.cc
  56. bias_op_gpu.h
  57. bias_op_test.cc
  58. bincount_op.cc
  59. bincount_op.h
  60. bincount_op_gpu.cu.cc
  61. bincount_op_test.cc
  62. broadcast_to_op.cc
  63. broadcast_to_op.h
  64. broadcast_to_op_gpu.cu.cc
  65. bucketize_op.cc
  66. bucketize_op.h
  67. bucketize_op_gpu.cu.cc
  68. BUILD
  69. candidate_sampler_ops.cc
  70. cast_op.cc
  71. cast_op.h
  72. cast_op_gpu.cu.cc
  73. cast_op_impl.h
  74. cast_op_impl_bfloat.cc
  75. cast_op_impl_bool.cc
  76. cast_op_impl_complex128.cc
  77. cast_op_impl_complex64.cc
  78. cast_op_impl_double.cc
  79. cast_op_impl_float.cc
  80. cast_op_impl_half.cc
  81. cast_op_impl_int16.cc
  82. cast_op_impl_int32.cc
  83. cast_op_impl_int64.cc
  84. cast_op_impl_int8.cc
  85. cast_op_impl_uint16.cc
  86. cast_op_impl_uint32.cc
  87. cast_op_impl_uint64.cc
  88. cast_op_impl_uint8.cc
  89. cast_op_test.cc
  90. check_numerics_op.cc
  91. check_numerics_op_gpu.cu.cc
  92. cholesky_grad.cc
  93. cholesky_op.cc
  94. clustering_ops.cc
  95. clustering_ops_test.cc
  96. collective_nccl.cc
  97. collective_nccl.h
  98. collective_nccl_broadcaster.cc
  99. collective_nccl_broadcaster.h
  100. collective_nccl_gatherer.cc
  101. collective_nccl_gatherer.h
  102. collective_nccl_reducer.cc
  103. collective_nccl_reducer.h
  104. collective_nccl_test.cc
  105. collective_ops.cc
  106. colorspace_op.cc
  107. colorspace_op.h
  108. colorspace_op_gpu.cu.cc
  109. colorspace_op_test.cc
  110. compare_and_bitpack_op.cc
  111. compare_and_bitpack_op.h
  112. compare_and_bitpack_op_gpu.cu.cc
  113. concat_lib.h
  114. concat_lib_cpu.cc
  115. concat_lib_cpu.h
  116. concat_lib_gpu.cc
  117. concat_lib_gpu.h
  118. concat_lib_gpu_impl.cu.cc
  119. concat_op.cc
  120. concat_op_test.cc
  121. conditional_accumulator.h
  122. conditional_accumulator_base.cc
  123. conditional_accumulator_base.h
  124. conditional_accumulator_base_op.cc
  125. conditional_accumulator_base_op.h
  126. conditional_accumulator_op.cc
  127. constant_op.cc
  128. constant_op.h
  129. constant_op_gpu.cu.cc
  130. constant_op_test.cc
  131. control_flow_ops.cc
  132. control_flow_ops.h
  133. control_flow_ops_test.cc
  134. conv_2d.h
  135. conv_2d_gpu.h
  136. conv_2d_gpu_double.cu.cc
  137. conv_2d_gpu_float.cu.cc
  138. conv_2d_gpu_half.cu.cc
  139. conv_2d_gpu_int.cu.cc
  140. conv_2d_gpu_uint16.cu.cc
  141. conv_2d_gpu_uint32.cu.cc
  142. conv_2d_gpu_uint64.cu.cc
  143. conv_2d_gpu_uint8.cu.cc
  144. conv_3d.h
  145. conv_grad_filter_ops.cc
  146. conv_grad_filter_ops_benchmark_test.cc
  147. conv_grad_input_ops.cc
  148. conv_grad_input_ops_benchmark_test.cc
  149. conv_grad_ops.cc
  150. conv_grad_ops.h
  151. conv_grad_ops_3d.cc
  152. conv_ops.cc
  153. conv_ops.h
  154. conv_ops_3d.cc
  155. conv_ops_benchmark_test.cc
  156. conv_ops_fused_double.cc
  157. conv_ops_fused_float.cc
  158. conv_ops_fused_half.cc
  159. conv_ops_fused_image_transform.cc
  160. conv_ops_fused_impl.h
  161. conv_ops_gpu.h
  162. conv_ops_gpu_2.cu.cc
  163. conv_ops_test.cc
  164. conv_ops_using_gemm.cc
  165. count_up_to_op.cc
  166. crop_and_resize_op.cc
  167. crop_and_resize_op.h
  168. crop_and_resize_op_benchmark_test.cc
  169. crop_and_resize_op_gpu.cu.cc
  170. crop_and_resize_op_test.cc
  171. cross_op.cc
  172. cross_op.h
  173. cross_op_gpu.cu.cc
  174. cross_op_test.cc
  175. ctc_decoder_ops.cc
  176. ctc_loss_op.cc
  177. cuda_solvers.cc
  178. cuda_solvers.h
  179. cuda_sparse.cc
  180. cuda_sparse.h
  181. cudnn_pooling_gpu.cc
  182. cudnn_pooling_gpu.h
  183. cudnn_rnn_ops.cc
  184. cwise_op_abs.cc
  185. cwise_op_acos.cc
  186. cwise_op_acosh.cc
  187. cwise_op_add_1.cc
  188. cwise_op_add_2.cc
  189. cwise_op_arg.cc
  190. cwise_op_asin.cc
  191. cwise_op_asinh.cc
  192. cwise_op_atan.cc
  193. cwise_op_atan2.cc
  194. cwise_op_atanh.cc
  195. cwise_op_bessel.cc
  196. cwise_op_bessel.cu.cc
  197. cwise_op_bitwise_and.cc
  198. cwise_op_bitwise_or.cc
  199. cwise_op_bitwise_xor.cc
  200. cwise_op_ceil.cc
  201. cwise_op_clip.cc
  202. cwise_op_clip.h
  203. cwise_op_clip_gpu.cu.cc
  204. cwise_op_complex.cc
  205. cwise_op_conj.cc
  206. cwise_op_cos.cc
  207. cwise_op_cosh.cc
  208. cwise_op_digamma.cc
  209. cwise_op_div.cc
  210. cwise_op_equal_to_1.cc
  211. cwise_op_equal_to_2.cc
  212. cwise_op_erf.cc
  213. cwise_op_erfc.cc
  214. cwise_op_exp.cc
  215. cwise_op_expm1.cc
  216. cwise_op_floor.cc
  217. cwise_op_floor_div.cc
  218. cwise_op_floor_mod.cc
  219. cwise_op_gpu_abs.cu.cc
  220. cwise_op_gpu_acos.cu.cc
  221. cwise_op_gpu_acosh.cu.cc
  222. cwise_op_gpu_add.cu.cc
  223. cwise_op_gpu_arg.cu.cc
  224. cwise_op_gpu_asin.cu.cc
  225. cwise_op_gpu_asinh.cu.cc
  226. cwise_op_gpu_atan.cu.cc
  227. cwise_op_gpu_atan2.cu.cc
  228. cwise_op_gpu_atanh.cu.cc
  229. cwise_op_gpu_bitwise_and.cu.cc
  230. cwise_op_gpu_bitwise_or.cu.cc
  231. cwise_op_gpu_bitwise_xor.cu.cc
  232. cwise_op_gpu_ceil.cu.cc
  233. cwise_op_gpu_complex.cu.cc
  234. cwise_op_gpu_conj.cu.cc
  235. cwise_op_gpu_cos.cu.cc
  236. cwise_op_gpu_cosh.cu.cc
  237. cwise_op_gpu_digamma.cu.cc
  238. cwise_op_gpu_div.cu.cc
  239. cwise_op_gpu_equal_to.cu.cc
  240. cwise_op_gpu_erf.cu.cc
  241. cwise_op_gpu_erfc.cu.cc
  242. cwise_op_gpu_exp.cu.cc
  243. cwise_op_gpu_expm1.cu.cc
  244. cwise_op_gpu_floor.cu.cc
  245. cwise_op_gpu_floor_div.cu.cc
  246. cwise_op_gpu_floor_mod.cu.cc
  247. cwise_op_gpu_greater.cu.cc
  248. cwise_op_gpu_greater_equal.cu.cc
  249. cwise_op_gpu_igammas_double.cu.cc
  250. cwise_op_gpu_igammas_float.cu.cc
  251. cwise_op_gpu_imag.cu.cc
  252. cwise_op_gpu_inverse.cu.cc
  253. cwise_op_gpu_invert.cu.cc
  254. cwise_op_gpu_isfinite.cu.cc
  255. cwise_op_gpu_isinf.cu.cc
  256. cwise_op_gpu_isnan.cu.cc
  257. cwise_op_gpu_left_shift.cu.cc
  258. cwise_op_gpu_less.cu.cc
  259. cwise_op_gpu_less_equal.cu.cc
  260. cwise_op_gpu_lgamma.cu.cc
  261. cwise_op_gpu_log.cu.cc
  262. cwise_op_gpu_log1p.cu.cc
  263. cwise_op_gpu_logical_and.cu.cc
  264. cwise_op_gpu_logical_not.cu.cc
  265. cwise_op_gpu_logical_or.cu.cc
  266. cwise_op_gpu_maximum.cu.cc
  267. cwise_op_gpu_minimum.cu.cc
  268. cwise_op_gpu_mod.cu.cc
  269. cwise_op_gpu_mul.cu.cc
  270. cwise_op_gpu_neg.cu.cc
  271. cwise_op_gpu_not_equal_to.cu.cc
  272. cwise_op_gpu_pow.cu.cc
  273. cwise_op_gpu_random_grad.cu.cc
  274. cwise_op_gpu_real.cu.cc
  275. cwise_op_gpu_right_shift.cu.cc
  276. cwise_op_gpu_rint.cu.cc
  277. cwise_op_gpu_round.cu.cc
  278. cwise_op_gpu_rsqrt.cu.cc
  279. cwise_op_gpu_select.cu.cc
  280. cwise_op_gpu_sigmoid.cu.cc
  281. cwise_op_gpu_sign.cu.cc
  282. cwise_op_gpu_sin.cu.cc
  283. cwise_op_gpu_sinh.cu.cc
  284. cwise_op_gpu_sqrt.cu.cc
  285. cwise_op_gpu_square.cu.cc
  286. cwise_op_gpu_squared_difference.cu.cc
  287. cwise_op_gpu_sub.cu.cc
  288. cwise_op_gpu_tan.cu.cc
  289. cwise_op_gpu_tanh.cu.cc
  290. cwise_op_gpu_xdivy.cu.cc
  291. cwise_op_gpu_xlogy.cu.cc
  292. cwise_op_gpu_zeta.cu.cc
  293. cwise_op_greater.cc
  294. cwise_op_greater_equal.cc
  295. cwise_op_igammas.cc
  296. cwise_op_imag.cc
  297. cwise_op_invert.cc
  298. cwise_op_isfinite.cc
  299. cwise_op_isinf.cc
  300. cwise_op_isnan.cc
  301. cwise_op_left_shift.cc
  302. cwise_op_less.cc
  303. cwise_op_less_equal.cc
  304. cwise_op_lgamma.cc
  305. cwise_op_log.cc
  306. cwise_op_log1p.cc
  307. cwise_op_logical_and.cc
  308. cwise_op_logical_not.cc
  309. cwise_op_logical_or.cc
  310. cwise_op_maximum.cc
  311. cwise_op_minimum.cc
  312. cwise_op_mod.cc
  313. cwise_op_mul_1.cc
  314. cwise_op_mul_2.cc
  315. cwise_op_neg.cc
  316. cwise_op_not_equal_to_1.cc
  317. cwise_op_not_equal_to_2.cc
  318. cwise_op_pow.cc
  319. cwise_op_random_grad.cc
  320. cwise_op_real.cc
  321. cwise_op_reciprocal.cc
  322. cwise_op_right_shift.cc
  323. cwise_op_rint.cc
  324. cwise_op_round.cc
  325. cwise_op_rsqrt.cc
  326. cwise_op_select.cc
  327. cwise_op_sigmoid.cc
  328. cwise_op_sign.cc
  329. cwise_op_sin.cc
  330. cwise_op_sinh.cc
  331. cwise_op_sqrt.cc
  332. cwise_op_square.cc
  333. cwise_op_squared_difference.cc
  334. cwise_op_sub.cc
  335. cwise_op_tan.cc
  336. cwise_op_tanh.cc
  337. cwise_op_xdivy.cc
  338. cwise_op_xlogy.cc
  339. cwise_op_zeta.cc
  340. cwise_ops.h
  341. cwise_ops_common.cc
  342. cwise_ops_common.h
  343. cwise_ops_gpu_common.cu.h
  344. cwise_ops_gpu_gradients.cu.h
  345. cwise_ops_gradients.h
  346. cwise_ops_sycl_common.h
  347. cwise_ops_test.cc
  348. data_format_ops.cc
  349. data_format_ops.h
  350. data_format_ops_gpu.cu.cc
  351. debug_ops.cc
  352. debug_ops.h
  353. debug_ops_test.cc
  354. decode_bmp_op.cc
  355. decode_compressed_op.cc
  356. decode_csv_op.cc
  357. decode_image_op.cc
  358. decode_padded_raw_op.cc
  359. decode_proto_op.cc
  360. decode_raw_op.cc
  361. decode_wav_op.cc
  362. decode_wav_op_test.cc
  363. deep_conv2d.cc
  364. deep_conv2d.h
  365. deep_conv2d_test.cc
  366. dense_update_functor.cc
  367. dense_update_functor.h
  368. dense_update_functor_gpu.cu.cc
  369. dense_update_ops.cc
  370. depthtospace_op.cc
  371. depthtospace_op.h
  372. depthtospace_op_gpu.cu.cc
  373. depthwise_conv_grad_op.cc
  374. depthwise_conv_op.cc
  375. depthwise_conv_op.h
  376. depthwise_conv_op_gpu.h
  377. depthwise_conv_op_gpu_double.cu.cc
  378. depthwise_conv_op_gpu_float.cu.cc
  379. depthwise_conv_op_gpu_half.cu.cc
  380. depthwise_conv_ops_test.cc
  381. dequantize_op.cc
  382. dequantize_op_test.cc
  383. deserialize_sparse_string_op.cc
  384. deserialize_sparse_variant_op.cc
  385. determinant_op.cc
  386. determinant_op.h
  387. determinant_op_gpu.cu.cc
  388. diag_op.cc
  389. diag_op.h
  390. diag_op_gpu.cu.cc
  391. diag_op_test.cc
  392. dilation_ops.cc
  393. dilation_ops.h
  394. dilation_ops_gpu.cu.cc
  395. draw_bounding_box_op.cc
  396. dynamic_partition_op.cc
  397. dynamic_partition_op_gpu.cu.cc
  398. dynamic_partition_op_test.cc
  399. dynamic_stitch_op.cc
  400. dynamic_stitch_op_gpu.cu.cc
  401. dynamic_stitch_op_test.cc
  402. edit_distance_op.cc
  403. eigen_activations.h
  404. eigen_activations_test.cc
  405. eigen_attention.h
  406. eigen_attention_test.cc
  407. eigen_backward_cuboid_convolutions.h
  408. eigen_backward_spatial_convolutions.h
  409. eigen_backward_spatial_convolutions_test.cc
  410. eigen_benchmark.h
  411. eigen_benchmark_cpu_test.cc
  412. eigen_contraction_kernel.cc
  413. eigen_contraction_kernel.h
  414. eigen_convolution_helpers.h
  415. eigen_cuboid_convolution.h
  416. eigen_mkldnn_contraction_kernel_test.cc
  417. eigen_pooling.h
  418. eigen_pooling_test.cc
  419. eigen_spatial_convolutions-inl.h
  420. eigen_spatial_convolutions.h
  421. eigen_spatial_convolutions_test.cc
  422. eigen_volume_patch.h
  423. einsum_op.cc
  424. einsum_op.h
  425. einsum_op_gpu.cu.cc
  426. encode_jpeg_op.cc
  427. encode_png_op.cc
  428. encode_proto_op.cc
  429. encode_wav_op.cc
  430. encode_wav_op_test.cc
  431. example_parsing_ops.cc
  432. example_parsing_ops_test.cc
  433. extract_image_patches_op.cc
  434. extract_image_patches_op.h
  435. extract_image_patches_op_gpu.cu.cc
  436. extract_jpeg_shape_op.cc
  437. extract_volume_patches_op.cc
  438. extract_volume_patches_op.h
  439. extract_volume_patches_op_gpu.cu.cc
  440. eye_functor.h
  441. eye_functor_gpu.cu.cc
  442. fact_op.cc
  443. fake_quant_ops.cc
  444. fake_quant_ops_functor.h
  445. fake_quant_ops_gpu.cu.cc
  446. fake_quant_ops_test.cc
  447. fft_ops.cc
  448. fifo_queue.cc
  449. fifo_queue.h
  450. fifo_queue_op.cc
  451. fill_functor.cc
  452. fill_functor.cu.cc
  453. fill_functor.h
  454. fingerprint_op.cc
  455. fingerprint_op_test.cc
  456. fixed_length_record_reader_op.cc
  457. fractional_avg_pool_op.cc
  458. fractional_max_pool_op.cc
  459. fractional_pool_common.cc
  460. fractional_pool_common.h
  461. function_ops.cc
  462. function_ops.h
  463. functional_ops.cc
  464. fused_batch_norm_ex_op_test.cc
  465. fused_batch_norm_op.cc
  466. fused_batch_norm_op.cu.cc
  467. fused_batch_norm_op.h
  468. fused_batch_norm_op_test.cc
  469. fused_eigen_output_kernels.cc
  470. fused_eigen_output_kernels.h
  471. gather_functor.cc
  472. gather_functor.h
  473. gather_functor_batched.cc
  474. gather_functor_batched.h
  475. gather_functor_batched_gpu.cu.cc
  476. gather_functor_batched_gpu.cu.h
  477. gather_functor_gpu.cu.cc
  478. gather_functor_gpu.cu.h
  479. gather_nd_op.cc
  480. gather_nd_op.h
  481. gather_nd_op_cpu_impl.h
  482. gather_nd_op_cpu_impl_0.cc
  483. gather_nd_op_cpu_impl_1.cc
  484. gather_nd_op_cpu_impl_2.cc
  485. gather_nd_op_cpu_impl_3.cc
  486. gather_nd_op_cpu_impl_4.cc
  487. gather_nd_op_cpu_impl_5.cc
  488. gather_nd_op_cpu_impl_6.cc
  489. gather_nd_op_cpu_impl_7.cc
  490. gather_nd_op_gpu.cu.cc
  491. gather_nd_op_test.cc
  492. gather_op.cc
  493. gather_op_test.cc
  494. gemm_functors.h
  495. generate_vocab_remapping_op.cc
  496. gpu_device_array.h
  497. gpu_device_array_gpu.h
  498. gpu_utils.cc
  499. gpu_utils.h
  500. guarantee_const_op.cc
  501. guarantee_const_op_test.cc
  502. hinge-loss.h
  503. histogram_op.cc
  504. histogram_op.h
  505. histogram_op_gpu.cu.cc
  506. host_constant_op.cc
  507. host_constant_op.h
  508. i_remote_fused_graph_executor.h
  509. i_remote_fused_graph_ops_definitions.cc
  510. i_remote_fused_graph_ops_definitions.h
  511. identity_n_op.cc
  512. identity_n_op.h
  513. identity_n_op_test.cc
  514. identity_op.cc
  515. identity_op.h
  516. identity_op_test.cc
  517. identity_reader_op.cc
  518. image_resizer_state.h
  519. immutable_constant_op.cc
  520. immutable_constant_op.h
  521. immutable_constant_op_test.cc
  522. in_topk_op.cc
  523. in_topk_op.h
  524. in_topk_op_gpu.cu.cc
  525. in_topk_op_test.cc
  526. initializable_lookup_table.cc
  527. initializable_lookup_table.h
  528. inplace_ops.cc
  529. inplace_ops_functor.h
  530. inplace_ops_functor_gpu.cu.cc
  531. kernel_platform_strings.h
  532. l2loss_op.cc
  533. l2loss_op.h
  534. l2loss_op_gpu.cu.cc
  535. linalg_ops_common.cc
  536. linalg_ops_common.h
  537. list_kernels.cc
  538. list_kernels.cu.cc
  539. list_kernels.h
  540. listdiff_op.cc
  541. lmdb_reader_op.cc
  542. load_and_remap_matrix_op.cc
  543. logging_ops.cc
  544. logging_ops.h
  545. logging_ops_test.cc
  546. logistic-loss.h
  547. lookup_table_init_op.cc
  548. lookup_table_init_op.h
  549. lookup_table_op.cc
  550. lookup_table_op.h
  551. lookup_util.cc
  552. lookup_util.h
  553. loss.h
  554. loss_test.cc
  555. lrn_op.cc
  556. lrn_op_test.cc
  557. lu_op.cc
  558. lu_op_gpu.cu.cc
  559. map_stage_op.cc
  560. matching_files_op.cc
  561. matmul_op.cc
  562. matmul_op.h
  563. matmul_op_fused.cc
  564. matmul_op_test.cc
  565. matrix_band_part_op.cc
  566. matrix_band_part_op.h
  567. matrix_band_part_op_gpu.cu.cc
  568. matrix_diag_op.cc
  569. matrix_diag_op.h
  570. matrix_diag_op_gpu.cu.cc
  571. matrix_exponential_op.cc
  572. matrix_inverse_op.cc
  573. matrix_logarithm_op.cc
  574. matrix_set_diag_op.cc
  575. matrix_set_diag_op.h
  576. matrix_set_diag_op_gpu.cu.cc
  577. matrix_solve_ls_op_complex128.cc
  578. matrix_solve_ls_op_complex64.cc
  579. matrix_solve_ls_op_double.cc
  580. matrix_solve_ls_op_float.cc
  581. matrix_solve_ls_op_impl.h
  582. matrix_solve_op.cc
  583. matrix_square_root_op.cc
  584. matrix_triangular_solve_op.cc
  585. maxpooling_op.cc
  586. maxpooling_op.h
  587. maxpooling_op_gpu.cu.cc
  588. maxpooling_op_gpu.h
  589. merge_v2_checkpoints_op_test.cc
  590. meta_support.cc
  591. meta_support.h
  592. mfcc.cc
  593. mfcc.h
  594. mfcc_dct.cc
  595. mfcc_dct.h
  596. mfcc_dct_test.cc
  597. mfcc_mel_filterbank.cc
  598. mfcc_mel_filterbank.h
  599. mfcc_mel_filterbank_test.cc
  600. mfcc_op.cc
  601. mfcc_op_test.cc
  602. mfcc_test.cc
  603. mirror_pad_op.cc
  604. mirror_pad_op.h
  605. mirror_pad_op_cpu_impl.h
  606. mirror_pad_op_cpu_impl_1.cc
  607. mirror_pad_op_cpu_impl_2.cc
  608. mirror_pad_op_cpu_impl_3.cc
  609. mirror_pad_op_cpu_impl_4.cc
  610. mirror_pad_op_cpu_impl_5.cc
  611. mirror_pad_op_gpu.cu.cc
  612. mkl_aggregate_ops.cc
  613. mkl_avgpooling_op.cc
  614. mkl_batch_matmul_op.cc
  615. mkl_concat_op.cc
  616. mkl_conv_grad_filter_ops.cc
  617. mkl_conv_grad_input_ops.cc
  618. mkl_conv_ops.cc
  619. mkl_conv_ops.h
  620. mkl_conv_ops_test.cc
  621. mkl_cwise_ops_common.cc
  622. mkl_dequantize_op.cc
  623. mkl_dequantize_op_test.cc
  624. mkl_fused_batch_norm_op.cc
  625. mkl_fused_ops_test.cc
  626. mkl_identity_op.cc
  627. mkl_input_conversion_op.cc
  628. mkl_lrn_op.cc
  629. mkl_matmul_op.cc
  630. mkl_maxpooling_op.cc
  631. mkl_pooling_ops_common.cc
  632. mkl_pooling_ops_common.h
  633. mkl_qmatmul_op.cc
  634. mkl_qmatmul_op_test.cc
  635. mkl_quantize_op.cc
  636. mkl_quantize_op_test.cc
  637. mkl_quantized_concat_op_test.cc
  638. mkl_quantized_conv_ops.h
  639. mkl_quantized_conv_ops_perchannel_test.cc
  640. mkl_quantized_conv_ops_test.cc
  641. mkl_quantized_pooling_ops_test.cc
  642. mkl_relu_op.cc
  643. mkl_requantization_range_per_channel_op.cc
  644. mkl_requantize_ops_test.cc
  645. mkl_requantize_per_channel_op.cc
  646. mkl_reshape_op.cc
  647. mkl_slice_op.cc
  648. mkl_softmax_op.cc
  649. mkl_tfconv_op.h
  650. mkl_transpose_op.cc
  651. multinomial_op.cc
  652. multinomial_op.h
  653. multinomial_op_gpu.cu.cc
  654. multinomial_op_test.cc
  655. mutex_ops.cc
  656. nccl_ops.cc
  657. nextafter_op.cc
  658. nextafter_op.h
  659. nextafter_op_gpu.cu.cc
  660. nn_ops_test.cc
  661. no_op.cc
  662. no_op.h
  663. non_max_suppression_op.cc
  664. non_max_suppression_op.cu.cc
  665. non_max_suppression_op.h
  666. non_max_suppression_op_gpu_test.cc
  667. non_max_suppression_op_test.cc
  668. nth_element_op.cc
  669. nth_element_op.h
  670. one_hot_op.cc
  671. one_hot_op.h
  672. one_hot_op_gpu.cu.cc
  673. ops_testutil.cc
  674. ops_testutil.h
  675. ops_testutil_test.cc
  676. ops_util.h
  677. ops_util_test.cc
  678. pack_op.cc
  679. pad_op.cc
  680. pad_op.h
  681. pad_op_gpu.cu.cc
  682. padding_fifo_queue.cc
  683. padding_fifo_queue.h
  684. padding_fifo_queue_op.cc
  685. parameterized_truncated_normal_op.cc
  686. parameterized_truncated_normal_op.h
  687. parameterized_truncated_normal_op_gpu.cu.cc
  688. parameterized_truncated_normal_op_test.cc
  689. parse_tensor_op.cc
  690. parse_tensor_test.cc
  691. partitioned_function_ops.cc
  692. partitioned_function_ops.h
  693. poisson-loss.h
  694. pooling_ops_3d.cc
  695. pooling_ops_3d.h
  696. pooling_ops_3d_gpu.cu.cc
  697. pooling_ops_3d_gpu.h
  698. pooling_ops_3d_sycl.h
  699. pooling_ops_common.cc
  700. pooling_ops_common.h
  701. pooling_ops_common_gpu.h
  702. population_count_op.cc
  703. population_count_op.h
  704. population_count_op_gpu.cu.cc
  705. priority_queue.cc
  706. priority_queue.h
  707. priority_queue_op.cc
  708. qr_op_complex128.cc
  709. qr_op_complex64.cc
  710. qr_op_double.cc
  711. qr_op_float.cc
  712. qr_op_impl.h
  713. quantization_utils.cc
  714. quantization_utils.h
  715. quantization_utils_test.cc
  716. quantize_and_dequantize_op.cc
  717. quantize_and_dequantize_op.h
  718. quantize_and_dequantize_op_gpu.cu.cc
  719. quantize_and_dequantize_op_test.cc
  720. quantize_down_and_shrink_range.cc
  721. quantize_down_and_shrink_range_op_test.cc
  722. quantize_op.cc
  723. quantize_op_test.cc
  724. quantized_activation_ops.cc
  725. quantized_activation_ops_test.cc
  726. quantized_add_op.cc
  727. quantized_add_op_test.cc
  728. quantized_batch_norm_op.cc
  729. quantized_batch_norm_op_test.cc
  730. quantized_bias_add_op.cc
  731. quantized_bias_add_op_test.cc
  732. quantized_concat_op.cc
  733. quantized_concat_op_test.cc
  734. quantized_conv_ops.cc
  735. quantized_conv_ops_test.cc
  736. quantized_instance_norm.cc
  737. quantized_instance_norm_test.cc
  738. quantized_matmul_op.cc
  739. quantized_matmul_op_test.cc
  740. quantized_mul_op.cc
  741. quantized_mul_op_test.cc
  742. quantized_pooling_ops.cc
  743. quantized_pooling_ops_test.cc
  744. quantized_reshape_op.cc
  745. quantized_reshape_op_test.cc
  746. quantized_resize_bilinear_op.cc
  747. quantized_resize_bilinear_op_test.cc
  748. queue_base.cc
  749. queue_base.h
  750. queue_op.cc
  751. queue_op.h
  752. queue_ops.cc
  753. ragged_gather_op.cc
  754. ragged_gather_op_test.cc
  755. ragged_range_op.cc
  756. ragged_range_op_test.cc
  757. ragged_tensor_from_variant_op.cc
  758. ragged_tensor_from_variant_op_test.cc
  759. ragged_tensor_to_sparse_kernel.cc
  760. ragged_tensor_to_sparse_kernel_test.cc
  761. ragged_tensor_to_tensor_op.cc
  762. ragged_tensor_to_tensor_op_test.cc
  763. ragged_tensor_to_variant_op.cc
  764. ragged_tensor_to_variant_op_test.cc
  765. random_binomial_op.cc
  766. random_binomial_op.h
  767. random_binomial_op_test.cc
  768. random_crop_op.cc
  769. random_op.cc
  770. random_op.h
  771. random_op_cpu.h
  772. random_op_gpu.cu.cc
  773. random_op_gpu.h
  774. random_op_test.cc
  775. random_poisson_op.cc
  776. random_poisson_op.h
  777. random_poisson_op_test.cc
  778. random_shuffle_op.cc
  779. random_shuffle_queue_op.cc
  780. range_sampler.cc
  781. range_sampler.h
  782. range_sampler_test.cc
  783. reader_ops.cc
  784. record_input_op.cc
  785. record_yielder.cc
  786. record_yielder.h
  787. reduce_join_op.cc
  788. reduction_gpu_kernels.cu.h
  789. reduction_ops.h
  790. reduction_ops_all.cc
  791. reduction_ops_any.cc
  792. reduction_ops_common.cc
  793. reduction_ops_common.h
  794. reduction_ops_common_gpu.h
  795. reduction_ops_euclidean.cc
  796. reduction_ops_gpu_bool.cu.cc
  797. reduction_ops_gpu_complex128.cu.cc
  798. reduction_ops_gpu_complex64.cu.cc
  799. reduction_ops_gpu_double.cu.cc
  800. reduction_ops_gpu_float.cu.cc
  801. reduction_ops_gpu_int.cu.cc
  802. reduction_ops_half_mean_sum.cu.cc
  803. reduction_ops_half_prod_max_min.cu.cc
  804. reduction_ops_max.cc
  805. reduction_ops_mean.cc
  806. reduction_ops_min.cc
  807. reduction_ops_prod.cc
  808. reduction_ops_sum.cc
  809. reduction_ops_test.cc
  810. redux_functor.h
  811. reference_gemm.h
  812. regex_full_match_op.cc
  813. regex_replace_op.cc
  814. regex_replace_op_test.cc
  815. relu_op.cc
  816. relu_op.h
  817. relu_op_functor.h
  818. relu_op_gpu.cu.cc
  819. remote_fused_graph_execute_op.cc
  820. remote_fused_graph_execute_op_test.cc
  821. remote_fused_graph_execute_op_test_utils.cc
  822. remote_fused_graph_execute_op_test_utils.h
  823. remote_fused_graph_execute_utils.cc
  824. remote_fused_graph_execute_utils.h
  825. remote_fused_graph_execute_utils_test.cc
  826. remote_fused_graph_rewriter_transform.cc
  827. remote_fused_graph_rewriter_transform_test.cc
  828. requantization_range_op.cc
  829. requantization_range_op_test.cc
  830. requantize.cc
  831. requantize_op_test.cc
  832. reshape_op.cc
  833. reshape_op.h
  834. reshape_util.cc
  835. reshape_util.h
  836. resize_area_op.cc
  837. resize_area_op_test.cc
  838. resize_bicubic_op.cc
  839. resize_bicubic_op_test.cc
  840. resize_bilinear_op.cc
  841. resize_bilinear_op.h
  842. resize_bilinear_op_gpu.cu.cc
  843. resize_bilinear_op_test.cc
  844. resize_nearest_neighbor_op.cc
  845. resize_nearest_neighbor_op.h
  846. resize_nearest_neighbor_op_gpu.cu.cc
  847. resize_nearest_neighbor_op_test.cc
  848. resize_op_benchmark_test.cc
  849. resource_variable_ops.cc
  850. resource_variable_ops.h
  851. restore_op.cc
  852. restore_op_test.cc
  853. restore_v2_op_test.cc
  854. reverse_op.cc
  855. reverse_op.h
  856. reverse_op_gpu.cu.cc
  857. reverse_op_test.cc
  858. reverse_sequence_op.cc
  859. reverse_sequence_op.h
  860. reverse_sequence_op_gpu.cu.cc
  861. roll_op.cc
  862. roll_op.h
  863. roll_op_gpu.cu.cc
  864. roll_op_test.cc
  865. rpc_op.cc
  866. sample_distorted_bounding_box_op.cc
  867. sampling_kernels.cc
  868. sampling_kernels.h
  869. sampling_kernels_test.cc
  870. save_op.cc
  871. save_op_test.cc
  872. save_restore_tensor.cc
  873. save_restore_tensor.h
  874. save_restore_v2_ops.cc
  875. save_v2_op_test.cc
  876. scale_and_translate_op.cc
  877. scale_and_translate_op.h
  878. scale_and_translate_op_test.cc
  879. scan_ops.cc
  880. scan_ops.h
  881. scan_ops_gpu.h
  882. scan_ops_gpu_double.cu.cc
  883. scan_ops_gpu_float.cu.cc
  884. scan_ops_gpu_half.cu.cc
  885. scan_ops_gpu_int.cu.cc
  886. scan_ops_test.cc
  887. scatter_functor.cc
  888. scatter_functor.h
  889. scatter_functor_gpu.cu.cc
  890. scatter_functor_gpu.cu.h
  891. scatter_nd_op.cc
  892. scatter_nd_op.h
  893. scatter_nd_op_cpu_impl.h
  894. scatter_nd_op_cpu_impl_0.cc
  895. scatter_nd_op_cpu_impl_1.cc
  896. scatter_nd_op_cpu_impl_2.cc
  897. scatter_nd_op_cpu_impl_3.cc
  898. scatter_nd_op_cpu_impl_4.cc
  899. scatter_nd_op_cpu_impl_5.cc
  900. scatter_nd_op_cpu_impl_6.cc
  901. scatter_nd_op_cpu_impl_7.cc
  902. scatter_nd_op_gpu.cu.cc
  903. scatter_nd_op_test.cc
  904. scatter_op.cc
  905. scatter_op_gpu.cu.cc
  906. scatter_op_test.cc
  907. scoped_allocator_ops.cc
  908. scoped_allocator_ops_test.cc
  909. sdca_internal.cc
  910. sdca_internal.h
  911. sdca_ops.cc
  912. sdca_ops_test.cc
  913. searchsorted_op.cc
  914. searchsorted_op.h
  915. searchsorted_op_gpu.cu.cc
  916. segment_reduction_ops.cc
  917. segment_reduction_ops.h
  918. segment_reduction_ops_gpu.cu.cc
  919. segment_reduction_ops_test.cc
  920. self_adjoint_eig_op.cc
  921. self_adjoint_eig_v2_op_complex128.cc
  922. self_adjoint_eig_v2_op_complex64.cc
  923. self_adjoint_eig_v2_op_double.cc
  924. self_adjoint_eig_v2_op_float.cc
  925. self_adjoint_eig_v2_op_gpu.cc
  926. self_adjoint_eig_v2_op_impl.h
  927. sendrecv_ops.cc
  928. sendrecv_ops.h
  929. sendrecv_ops_test.cc
  930. sequence_ops.cc
  931. sequence_ops_test.cc
  932. serialize_sparse_op.cc
  933. session_ops.cc
  934. set_kernels.cc
  935. shape_ops.cc
  936. shape_ops.h
  937. slice_op.cc
  938. slice_op.h
  939. slice_op_cpu_impl.h
  940. slice_op_cpu_impl_1.cc
  941. slice_op_cpu_impl_2.cc
  942. slice_op_cpu_impl_3.cc
  943. slice_op_cpu_impl_4.cc
  944. slice_op_cpu_impl_5.cc
  945. slice_op_cpu_impl_6.cc
  946. slice_op_cpu_impl_7.cc
  947. slice_op_gpu.cu.cc
  948. slice_op_test.cc
  949. smooth-hinge-loss.h
  950. snapshot_op.cc
  951. snapshot_op.h
  952. snapshot_op_gpu.cu.cc
  953. softmax_op.cc
  954. softmax_op_functor.h
  955. softmax_op_gpu.cu.cc
  956. softplus_op.cc
  957. softplus_op.h
  958. softplus_op_gpu.cu.cc
  959. softsign_op.cc
  960. softsign_op.h
  961. softsign_op_gpu.cu.cc
  962. spacetobatch_benchmark_test.cc
  963. spacetobatch_functor.cc
  964. spacetobatch_functor.h
  965. spacetobatch_functor_gpu.cu.cc
  966. spacetobatch_op.cc
  967. spacetodepth_op.cc
  968. spacetodepth_op.h
  969. spacetodepth_op_gpu.cu.cc
  970. sparse_add_grad_op.cc
  971. sparse_add_op.cc
  972. sparse_add_op_test.cc
  973. sparse_concat_op.cc
  974. sparse_conditional_accumulator.h
  975. sparse_conditional_accumulator_op.cc
  976. sparse_cross_op.cc
  977. sparse_dense_binary_op_shared.cc
  978. sparse_dense_binary_op_shared_test.cc
  979. sparse_fill_empty_rows_op.cc
  980. sparse_matmul_op.cc
  981. sparse_matmul_op.h
  982. sparse_matmul_op_test.cc
  983. sparse_reduce_op.cc
  984. sparse_reduce_sum_op_test.cc
  985. sparse_reorder_op.cc
  986. sparse_reshape_op.cc
  987. sparse_slice_grad_op.cc
  988. sparse_slice_op.cc
  989. sparse_softmax_op.cc
  990. sparse_sparse_binary_op_shared.cc
  991. sparse_split_op.cc
  992. sparse_tensor_dense_add_op.cc
  993. sparse_tensor_dense_add_op.h
  994. sparse_tensor_dense_matmul_op.cc
  995. sparse_tensor_dense_matmul_op.h
  996. sparse_tensor_dense_matmul_op_gpu.cu.cc
  997. sparse_tensor_dense_matmul_op_test.cc
  998. sparse_tensors_map_ops.cc
  999. sparse_to_dense_op.cc
  1000. sparse_to_dense_op_test.cc
  1001. sparse_utils.cc
  1002. sparse_utils.h
  1003. sparse_utils_test.cc
  1004. sparse_xent_op.cc
  1005. sparse_xent_op.h
  1006. sparse_xent_op_gpu.cu.cc
  1007. sparse_xent_op_test.cc
  1008. spectrogram.cc
  1009. spectrogram.h
  1010. spectrogram_convert_test_data.cc
  1011. spectrogram_op.cc
  1012. spectrogram_op_test.cc
  1013. spectrogram_test.cc
  1014. spectrogram_test_utils.cc
  1015. spectrogram_test_utils.h
  1016. split_lib.h
  1017. split_lib_cpu.cc
  1018. split_lib_gpu.cu.cc
  1019. split_lib_gpu.h
  1020. split_op.cc
  1021. split_op_test.cc
  1022. split_v_op.cc
  1023. split_v_op_test.cc
  1024. squared-loss.h
  1025. stack.cc
  1026. stack.h
  1027. stack_ops.cc
  1028. stage_op.cc
  1029. stateful_random_ops.cc
  1030. stateful_random_ops.h
  1031. stateful_random_ops_cpu_gpu.h
  1032. stateful_random_ops_gpu.cu.cc
  1033. stateless_random_ops.cc
  1034. stateless_random_ops.h
  1035. strided_slice_op.cc
  1036. strided_slice_op.h
  1037. strided_slice_op_define_grad.cc
  1038. strided_slice_op_gpu_bool.cu.cc
  1039. strided_slice_op_gpu_complex.cu.cc
  1040. strided_slice_op_gpu_impl.h
  1041. strided_slice_op_gpu_int.cu.cc
  1042. strided_slice_op_gpu_number_types.cu.cc
  1043. strided_slice_op_impl.h
  1044. strided_slice_op_inst_0.cc
  1045. strided_slice_op_inst_1.cc
  1046. strided_slice_op_inst_2.cc
  1047. strided_slice_op_inst_3.cc
  1048. strided_slice_op_inst_4.cc
  1049. strided_slice_op_inst_5.cc
  1050. strided_slice_op_inst_6.cc
  1051. strided_slice_op_inst_7.cc
  1052. strided_slice_op_test.cc
  1053. string_format_op.cc
  1054. string_format_op_test.cc
  1055. string_join_op.cc
  1056. string_length_op.cc
  1057. string_lower_op.cc
  1058. string_ngrams_op.cc
  1059. string_ngrams_op_test.cc
  1060. string_split_op.cc
  1061. string_split_op_test.cc
  1062. string_strip_op.cc
  1063. string_to_hash_bucket_op.cc
  1064. string_to_hash_bucket_op.h
  1065. string_to_number_op.cc
  1066. string_upper_op.cc
  1067. string_util.cc
  1068. string_util.h
  1069. substr_op.cc
  1070. substr_op_test.cc
  1071. summary_audio_op.cc
  1072. summary_audio_op_test.cc
  1073. summary_image_op.cc
  1074. summary_image_op_test.cc
  1075. summary_interface.h
  1076. summary_kernels.cc
  1077. summary_op.cc
  1078. summary_op_test.cc
  1079. summary_tensor_op.cc
  1080. summary_tensor_op_test.cc
  1081. svd_op_complex128.cc
  1082. svd_op_complex64.cc
  1083. svd_op_double.cc
  1084. svd_op_float.cc
  1085. svd_op_gpu.cu.cc
  1086. svd_op_impl.h
  1087. tensor_array.cc
  1088. tensor_array.h
  1089. tensor_array_ops.cc
  1090. tensor_flag_utils.cc
  1091. tensor_flag_utils.h
  1092. tensor_flag_utils_test.cc
  1093. text_line_reader_op.cc
  1094. tf_record_reader_op.cc
  1095. tile_functor.h
  1096. tile_functor_cpu.cc
  1097. tile_functor_gpu.h
  1098. tile_functor_gpu_bool.cu.cc
  1099. tile_functor_gpu_complex128.cu.cc
  1100. tile_functor_gpu_complex64.cu.cc
  1101. tile_functor_gpu_double.cu.cc
  1102. tile_functor_gpu_float.cu.cc
  1103. tile_functor_gpu_half.cu.cc
  1104. tile_functor_gpu_int16.cu.cc
  1105. tile_functor_gpu_int32.cu.cc
  1106. tile_functor_gpu_int64.cu.cc
  1107. tile_ops.cc
  1108. tile_ops_cpu_impl.h
  1109. tile_ops_cpu_impl_1.cc
  1110. tile_ops_cpu_impl_2.cc
  1111. tile_ops_cpu_impl_3.cc
  1112. tile_ops_cpu_impl_4.cc
  1113. tile_ops_cpu_impl_5.cc
  1114. tile_ops_cpu_impl_6.cc
  1115. tile_ops_cpu_impl_7.cc
  1116. tile_ops_gpu_impl.h
  1117. tile_ops_gpu_impl_1.cu.cc
  1118. tile_ops_gpu_impl_2.cu.cc
  1119. tile_ops_gpu_impl_3.cu.cc
  1120. tile_ops_gpu_impl_4.cu.cc
  1121. tile_ops_gpu_impl_5.cu.cc
  1122. tile_ops_gpu_impl_6.cu.cc
  1123. tile_ops_gpu_impl_7.cu.cc
  1124. tile_ops_gpu_impl_8.cu.cc
  1125. tile_ops_impl.h
  1126. topk_op.cc
  1127. topk_op.h
  1128. topk_op_gpu.h
  1129. topk_op_gpu_double.cu.cc
  1130. topk_op_gpu_float.cu.cc
  1131. topk_op_gpu_half.cu.cc
  1132. topk_op_gpu_int16.cu.cc
  1133. topk_op_gpu_int32.cu.cc
  1134. topk_op_gpu_int64.cu.cc
  1135. topk_op_gpu_int8.cu.cc
  1136. topk_op_gpu_uint16.cu.cc
  1137. topk_op_gpu_uint8.cu.cc
  1138. training_op_helpers.cc
  1139. training_op_helpers.h
  1140. training_ops.cc
  1141. training_ops.h
  1142. training_ops_gpu.cu.cc
  1143. training_ops_test.cc
  1144. transpose_functor.h
  1145. transpose_functor_cpu.cc
  1146. transpose_functor_gpu.cu.cc
  1147. transpose_op.cc
  1148. transpose_op.h
  1149. transpose_util_test.cc
  1150. tridiagonal_matmul_op.cc
  1151. tridiagonal_matmul_op_gpu.cu.cc
  1152. tridiagonal_solve_op.cc
  1153. tridiagonal_solve_op_gpu.cu.cc
  1154. typed_conditional_accumulator_base.h
  1155. typed_queue.h
  1156. unary_ops_composition.cc
  1157. unary_ops_composition_test.cc
  1158. unicode_ops.cc
  1159. unicode_script_op.cc
  1160. unique_op.cc
  1161. unique_op_test.cc
  1162. unpack_op.cc
  1163. unravel_index_op.cc
  1164. unsorted_segment_join_op.cc
  1165. variable_ops.cc
  1166. variable_ops.h
  1167. variable_ops_test.cc
  1168. where_op.cc
  1169. where_op.h
  1170. where_op_gpu.cu.h
  1171. where_op_gpu_impl_1.cu.cc
  1172. where_op_gpu_impl_2.cu.cc
  1173. where_op_gpu_impl_3.cu.cc
  1174. where_op_gpu_impl_4.cu.cc
  1175. where_op_gpu_impl_5.cu.cc
  1176. where_op_gpu_impl_6.cu.cc
  1177. where_op_gpu_impl_7.cu.cc
  1178. where_op_gpu_impl_8.cu.cc
  1179. whole_file_read_ops.cc
  1180. winograd_transform.h
  1181. word2vec_kernels.cc
  1182. xent_op.cc
  1183. xent_op.h
  1184. xent_op_gpu.cu.cc
  1185. xent_op_test.cc
  1186. xsmm_conv2d.cc
  1187. xsmm_conv2d.h
  1188. xsmm_conv2d_test.cc