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