Home
last modified time | relevance | path

Searched refs:CheckDatasetOutputShapes (Results 1 – 20 of 20) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/data/
Dmap_dataset_op_test.cc130 TF_ASSERT_OK(CheckDatasetOutputShapes({PartialTensorShape({})})); in TEST_F()
Drange_dataset_op_test.cc86 TF_ASSERT_OK(CheckDatasetOutputShapes({PartialTensorShape({})})); in DATASET_OUTPUT_DTYPES_TEST_P()
Dskip_dataset_op_test.cc156 TF_ASSERT_OK(CheckDatasetOutputShapes({PartialTensorShape({})})); in TEST_F()
Dflat_map_dataset_op_test.cc155 TF_ASSERT_OK(CheckDatasetOutputShapes(dataset_params.output_shapes())); in TEST_F()
Dprefetch_dataset_op_test.cc255 TF_ASSERT_OK(CheckDatasetOutputShapes(dataset_params.output_shapes())); in TEST_F()
Dtext_line_dataset_op_test.cc189 TF_ASSERT_OK(CheckDatasetOutputShapes({PartialTensorShape({})})); in TEST_F()
Dtf_record_dataset_op_test.cc217 TF_ASSERT_OK(CheckDatasetOutputShapes({PartialTensorShape({})})); in TEST_F()
Dshard_dataset_op_test.cc248 TF_ASSERT_OK(CheckDatasetOutputShapes({PartialTensorShape({})})); in TEST_F()
Dfixed_length_record_dataset_op_test.cc215 TF_ASSERT_OK(CheckDatasetOutputShapes({PartialTensorShape({})})); in TEST_F()
Dtensor_dataset_op_test.cc172 TF_ASSERT_OK(CheckDatasetOutputShapes(dataset_params.output_shapes())); in TEST_F()
Dparallel_map_dataset_op_test.cc326 TF_ASSERT_OK(CheckDatasetOutputShapes({PartialTensorShape({})})); in TEST_F()
Dshuffle_dataset_op_test.cc407 TF_ASSERT_OK(CheckDatasetOutputShapes(dataset_params.output_shapes())); in TEST_F()
Dwindow_dataset_op_test.cc352 TF_ASSERT_OK(CheckDatasetOutputShapes(dataset_params.output_shapes())); in TEST_F()
Ddataset_test_base.h613 Status CheckDatasetOutputShapes(
896 TF_ASSERT_OK(CheckDatasetOutputShapes(test_case.expected_output_shapes)); \
Dparallel_interleave_dataset_op_test.cc631 TF_ASSERT_OK(CheckDatasetOutputShapes({PartialTensorShape({1})})); in TEST_F()
Ddataset_test_base.cc691 Status DatasetOpsTestBase::CheckDatasetOutputShapes( in CheckDatasetOutputShapes() function in tensorflow::data::DatasetOpsTestBase
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dassert_next_dataset_op_test.cc166 TF_ASSERT_OK(CheckDatasetOutputShapes({PartialTensorShape({})})); in TEST_F()
Dunique_dataset_op_test.cc203 TF_ASSERT_OK(CheckDatasetOutputShapes({PartialTensorShape({1})})); in TEST_F()
Ddirected_interleave_dataset_op_test.cc279 TF_ASSERT_OK(CheckDatasetOutputShapes({PartialTensorShape({})})); in TEST_F()
Dparallel_interleave_dataset_op_test.cc433 TF_ASSERT_OK(CheckDatasetOutputShapes({PartialTensorShape({1})})); in TEST_F()