Searched refs:CheckIteratorOutputShapes (Results 1 – 19 of 19) sorted by relevance
154 TF_ASSERT_OK(CheckIteratorOutputShapes({PartialTensorShape({})})); in TEST_F()
113 TF_ASSERT_OK(CheckIteratorOutputShapes({PartialTensorShape({})})); in ITERATOR_OUTPUT_DTYPES_TEST_P()
184 TF_ASSERT_OK(CheckIteratorOutputShapes({PartialTensorShape({})})); in TEST_F()
173 TF_ASSERT_OK(CheckIteratorOutputShapes(dataset_params.output_shapes())); in TEST_F()
283 TF_ASSERT_OK(CheckIteratorOutputShapes(dataset_params.output_shapes())); in TEST_F()
207 TF_ASSERT_OK(CheckIteratorOutputShapes({PartialTensorShape({})})); in TEST_F()
235 TF_ASSERT_OK(CheckIteratorOutputShapes({PartialTensorShape({})})); in TEST_F()
280 TF_ASSERT_OK(CheckIteratorOutputShapes({PartialTensorShape({})})); in TEST_F()
233 TF_ASSERT_OK(CheckIteratorOutputShapes({PartialTensorShape({})})); in TEST_F()
190 TF_ASSERT_OK(CheckIteratorOutputShapes(dataset_params.output_shapes())); in TEST_F()
357 TF_ASSERT_OK(CheckIteratorOutputShapes({PartialTensorShape({})})); in TEST_F()
441 TF_ASSERT_OK(CheckIteratorOutputShapes(dataset_params.output_shapes())); in TEST_F()
391 TF_ASSERT_OK(CheckIteratorOutputShapes(dataset_params.output_shapes())); in TEST_F()
624 Status CheckIteratorOutputShapes(954 TF_ASSERT_OK(CheckIteratorOutputShapes(test_case.expected_output_shapes)); \
671 TF_ASSERT_OK(CheckIteratorOutputShapes({PartialTensorShape({1})})); in TEST_F()
710 Status DatasetOpsTestBase::CheckIteratorOutputShapes( in CheckIteratorOutputShapes() function in tensorflow::data::DatasetOpsTestBase
297 TF_ASSERT_OK(CheckIteratorOutputShapes({PartialTensorShape({})})); in TEST_F()
463 TF_ASSERT_OK(CheckIteratorOutputShapes({PartialTensorShape({1})})); in TEST_F()