Home
last modified time | relevance | path

Searched refs:CheckIteratorOutputShapes (Results 1 – 19 of 19) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/data/
Dmap_dataset_op_test.cc154 TF_ASSERT_OK(CheckIteratorOutputShapes({PartialTensorShape({})})); in TEST_F()
Drange_dataset_op_test.cc113 TF_ASSERT_OK(CheckIteratorOutputShapes({PartialTensorShape({})})); in ITERATOR_OUTPUT_DTYPES_TEST_P()
Dskip_dataset_op_test.cc184 TF_ASSERT_OK(CheckIteratorOutputShapes({PartialTensorShape({})})); in TEST_F()
Dflat_map_dataset_op_test.cc173 TF_ASSERT_OK(CheckIteratorOutputShapes(dataset_params.output_shapes())); in TEST_F()
Dprefetch_dataset_op_test.cc283 TF_ASSERT_OK(CheckIteratorOutputShapes(dataset_params.output_shapes())); in TEST_F()
Dtext_line_dataset_op_test.cc207 TF_ASSERT_OK(CheckIteratorOutputShapes({PartialTensorShape({})})); in TEST_F()
Dtf_record_dataset_op_test.cc235 TF_ASSERT_OK(CheckIteratorOutputShapes({PartialTensorShape({})})); in TEST_F()
Dshard_dataset_op_test.cc280 TF_ASSERT_OK(CheckIteratorOutputShapes({PartialTensorShape({})})); in TEST_F()
Dfixed_length_record_dataset_op_test.cc233 TF_ASSERT_OK(CheckIteratorOutputShapes({PartialTensorShape({})})); in TEST_F()
Dtensor_dataset_op_test.cc190 TF_ASSERT_OK(CheckIteratorOutputShapes(dataset_params.output_shapes())); in TEST_F()
Dparallel_map_dataset_op_test.cc357 TF_ASSERT_OK(CheckIteratorOutputShapes({PartialTensorShape({})})); in TEST_F()
Dshuffle_dataset_op_test.cc441 TF_ASSERT_OK(CheckIteratorOutputShapes(dataset_params.output_shapes())); in TEST_F()
Dwindow_dataset_op_test.cc391 TF_ASSERT_OK(CheckIteratorOutputShapes(dataset_params.output_shapes())); in TEST_F()
Ddataset_test_base.h624 Status CheckIteratorOutputShapes(
954 TF_ASSERT_OK(CheckIteratorOutputShapes(test_case.expected_output_shapes)); \
Dparallel_interleave_dataset_op_test.cc671 TF_ASSERT_OK(CheckIteratorOutputShapes({PartialTensorShape({1})})); in TEST_F()
Ddataset_test_base.cc710 Status DatasetOpsTestBase::CheckIteratorOutputShapes( in CheckIteratorOutputShapes() function in tensorflow::data::DatasetOpsTestBase
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dassert_next_dataset_op_test.cc184 TF_ASSERT_OK(CheckIteratorOutputShapes({PartialTensorShape({})})); in TEST_F()
Ddirected_interleave_dataset_op_test.cc297 TF_ASSERT_OK(CheckIteratorOutputShapes({PartialTensorShape({})})); in TEST_F()
Dparallel_interleave_dataset_op_test.cc463 TF_ASSERT_OK(CheckIteratorOutputShapes({PartialTensorShape({1})})); in TEST_F()