Searched refs:CheckIteratorOutputShapes (Results 1 – 21 of 21) sorted by relevance
125 TF_ASSERT_OK(CheckIteratorOutputShapes({PartialTensorShape({})})); in TEST_F()
154 TF_ASSERT_OK(CheckIteratorOutputShapes({PartialTensorShape({})})); in TEST_F()
113 TF_ASSERT_OK(CheckIteratorOutputShapes({PartialTensorShape({})})); in ITERATOR_OUTPUT_DTYPES_TEST_P()
185 TF_ASSERT_OK(CheckIteratorOutputShapes({PartialTensorShape({})})); in TEST_F()
192 TF_ASSERT_OK(CheckIteratorOutputShapes(dataset_params.output_shapes())); in TEST_F()
282 TF_ASSERT_OK(CheckIteratorOutputShapes(dataset_params.output_shapes())); in TEST_F()
208 TF_ASSERT_OK(CheckIteratorOutputShapes({PartialTensorShape({})})); in TEST_F()
280 TF_ASSERT_OK(CheckIteratorOutputShapes({PartialTensorShape({})})); in TEST_F()
236 TF_ASSERT_OK(CheckIteratorOutputShapes({PartialTensorShape({})})); in TEST_F()
234 TF_ASSERT_OK(CheckIteratorOutputShapes({PartialTensorShape({})})); in TEST_F()
196 TF_ASSERT_OK(CheckIteratorOutputShapes(dataset_params.output_shapes())); in TEST_F()
356 TF_ASSERT_OK(CheckIteratorOutputShapes({PartialTensorShape({})})); in TEST_F()
446 TF_ASSERT_OK(CheckIteratorOutputShapes(dataset_params.output_shapes())); in TEST_F()
396 TF_ASSERT_OK(CheckIteratorOutputShapes(dataset_params.output_shapes())); in TEST_F()
787 TF_ASSERT_OK(CheckIteratorOutputShapes({PartialTensorShape({1})})); in TEST_F()
184 TF_ASSERT_OK(CheckIteratorOutputShapes({PartialTensorShape({})})); in TEST_F()
239 TF_ASSERT_OK(CheckIteratorOutputShapes({PartialTensorShape({})})); in TEST_F()
379 TF_ASSERT_OK(CheckIteratorOutputShapes({PartialTensorShape({})})); in TEST_F()
463 TF_ASSERT_OK(CheckIteratorOutputShapes({PartialTensorShape({1})})); in TEST_F()
673 Status CheckIteratorOutputShapes(1007 TF_ASSERT_OK(CheckIteratorOutputShapes(test_case.expected_output_shapes)); \
725 Status DatasetOpsTestBase::CheckIteratorOutputShapes( in CheckIteratorOutputShapes() function in tensorflow::data::DatasetOpsTestBase