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