Home
last modified time | relevance | path

Searched refs:CheckDatasetCardinality (Results 1 – 11 of 11) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/
H A Doptions_dataset_op_test.cc113 TF_ASSERT_OK(CheckDatasetCardinality(4)); in TEST_F()
H A Dflat_map_dataset_op_test.cc180 TF_ASSERT_OK(CheckDatasetCardinality(kUnknownCardinality)); in TEST_F()
H A Dtext_line_dataset_op_test.cc196 TF_ASSERT_OK(CheckDatasetCardinality(kUnknownCardinality)); in TEST_F()
H A Dtf_record_dataset_op_test.cc224 TF_ASSERT_OK(CheckDatasetCardinality(kUnknownCardinality)); in TEST_F()
H A Dfixed_length_record_dataset_op_test.cc222 TF_ASSERT_OK(CheckDatasetCardinality(kUnknownCardinality)); in TEST_F()
H A Dtensor_dataset_op_test.cc184 TF_ASSERT_OK(CheckDatasetCardinality(1)); in TEST_F()
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/experimental/
H A Dassert_next_dataset_op_test.cc172 TF_ASSERT_OK(CheckDatasetCardinality(/*expected_cardinality=*/3)); in TEST_F()
H A Dassert_prev_dataset_op_test.cc227 TF_ASSERT_OK(CheckDatasetCardinality(/*expected_cardinality=*/3)); in TEST_F()
H A Ddirected_interleave_dataset_op_test.cc367 TF_ASSERT_OK(CheckDatasetCardinality(kUnknownCardinality)); in TEST_F()
/aosp_15_r20/external/tensorflow/tensorflow/core/data/
H A Ddataset_test_base.h663 Status CheckDatasetCardinality(int expected_cardinality);
969 TF_ASSERT_OK(CheckDatasetCardinality(test_case.expected_cardinality)); \
H A Ddataset_test_base.cc706 Status DatasetOpsTestBase::CheckDatasetCardinality(int expected_cardinality) { in CheckDatasetCardinality() function in tensorflow::data::DatasetOpsTestBase