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