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