Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/data/
Doptions_dataset_op_test.cc113 TF_ASSERT_OK(CheckDatasetCardinality(4)); in TEST_F()
Dflat_map_dataset_op_test.cc179 TF_ASSERT_OK(CheckDatasetCardinality(kUnknownCardinality)); in TEST_F()
Dtext_line_dataset_op_test.cc195 TF_ASSERT_OK(CheckDatasetCardinality(kUnknownCardinality)); in TEST_F()
Dtf_record_dataset_op_test.cc223 TF_ASSERT_OK(CheckDatasetCardinality(kUnknownCardinality)); in TEST_F()
Dfixed_length_record_dataset_op_test.cc221 TF_ASSERT_OK(CheckDatasetCardinality(kUnknownCardinality)); in TEST_F()
Dtensor_dataset_op_test.cc183 TF_ASSERT_OK(CheckDatasetCardinality(1)); in TEST_F()
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dassert_next_dataset_op_test.cc172 TF_ASSERT_OK(CheckDatasetCardinality(/*expected_cardinality=*/3)); in TEST_F()
Ddirected_interleave_dataset_op_test.cc356 TF_ASSERT_OK(CheckDatasetCardinality(kUnknownCardinality)); in TEST_F()
/external/tensorflow/tensorflow/core/data/
Ddataset_test_base.h653 Status CheckDatasetCardinality(int expected_cardinality);
959 TF_ASSERT_OK(CheckDatasetCardinality(test_case.expected_cardinality)); \
Ddataset_test_base.cc696 Status DatasetOpsTestBase::CheckDatasetCardinality(int expected_cardinality) { in CheckDatasetCardinality() function in tensorflow::data::DatasetOpsTestBase