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