Home
last modified time | relevance | path

Searched refs:CheckDatasetOutputDtypes (Results 1 – 25 of 29) sorted by relevance

12

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/
H A Doptions_dataset_op_test.cc101 TF_ASSERT_OK(CheckDatasetOutputDtypes({DT_INT64})); in TEST_F()
H A Dtake_dataset_op_test.cc97 TF_ASSERT_OK(CheckDatasetOutputDtypes({DT_INT64})); in TEST_F()
H A Dmap_dataset_op_test.cc124 TF_ASSERT_OK(CheckDatasetOutputDtypes({DT_INT64})); in TEST_F()
H A Dskip_dataset_op_test.cc151 TF_ASSERT_OK(CheckDatasetOutputDtypes({DT_INT64})); in TEST_F()
H A Dflat_map_dataset_op_test.cc168 TF_ASSERT_OK(CheckDatasetOutputDtypes(dataset_params.output_dtypes())); in TEST_F()
H A Dprefetch_dataset_op_test.cc248 TF_ASSERT_OK(CheckDatasetOutputDtypes({DT_INT64})); in TEST_F()
H A Dtext_line_dataset_op_test.cc184 TF_ASSERT_OK(CheckDatasetOutputDtypes({DT_STRING})); in TEST_F()
H A Dtf_record_dataset_op_test.cc212 TF_ASSERT_OK(CheckDatasetOutputDtypes({DT_STRING})); in TEST_F()
H A Dshard_dataset_op_test.cc242 TF_ASSERT_OK(CheckDatasetOutputDtypes({DT_INT64})); in TEST_F()
H A Dfilter_dataset_op_test.cc201 TF_ASSERT_OK(CheckDatasetOutputDtypes({DT_INT64})); in TEST_F()
H A Dfixed_length_record_dataset_op_test.cc210 TF_ASSERT_OK(CheckDatasetOutputDtypes({DT_STRING})); in TEST_F()
H A Dbatch_dataset_op_test.cc175 TF_ASSERT_OK(CheckDatasetOutputDtypes({DT_INT64})); in TEST_F()
H A Dparallel_batch_dataset_op_test.cc316 TF_ASSERT_OK(CheckDatasetOutputDtypes({DT_INT64})); in TEST_F()
H A Dcache_dataset_ops_test.cc223 TF_ASSERT_OK(CheckDatasetOutputDtypes({DT_INT64})); in TEST_F()
H A Dparallel_filter_dataset_op_test.cc334 TF_ASSERT_OK(CheckDatasetOutputDtypes({DT_INT64})); in TEST_F()
H A Dtensor_dataset_op_test.cc172 TF_ASSERT_OK(CheckDatasetOutputDtypes(dataset_params.output_dtypes())); in TEST_F()
H A Dparallel_map_dataset_op_test.cc319 TF_ASSERT_OK(CheckDatasetOutputDtypes({DT_INT64})); in TEST_F()
/aosp_15_r20/external/tensorflow/tensorflow/core/data/
H A Ddataset_test_base.h656 Status CheckDatasetOutputDtypes(const DataTypeVector& expected_output_dtypes);
929 TF_ASSERT_OK(CheckDatasetOutputDtypes(test_case.expected_output_dtypes)); \
988 TF_ASSERT_OK(CheckDatasetOutputDtypes(test_case.expected_output_dtypes)); \
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/experimental/
H A Dassert_next_dataset_op_test.cc160 TF_ASSERT_OK(CheckDatasetOutputDtypes({DT_INT64})); in TEST_F()
H A Dunique_dataset_op_test.cc198 TF_ASSERT_OK(CheckDatasetOutputDtypes({DT_INT64})); in TEST_F()
H A Dassert_prev_dataset_op_test.cc215 TF_ASSERT_OK(CheckDatasetOutputDtypes({DT_INT64})); in TEST_F()
H A Dio_ops_test.cc197 TF_ASSERT_OK(CheckDatasetOutputDtypes(dataset_params.output_dtypes())); in TEST_F()
H A Dmap_and_batch_dataset_op_test.cc288 TF_ASSERT_OK(CheckDatasetOutputDtypes({DT_INT64})); in TEST_F()
H A Ddirected_interleave_dataset_op_test.cc355 TF_ASSERT_OK(CheckDatasetOutputDtypes({DT_INT64})); in TEST_F()
H A Dparallel_interleave_dataset_op_test.cc427 TF_ASSERT_OK(CheckDatasetOutputDtypes({DT_INT64})); in TEST_F()

12