Searched refs:VerifyShapesCompatible (Results 1 – 12 of 12) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | dataset_utils.h | 57 Status VerifyShapesCompatible(const std::vector<PartialTensorShape>& expected,
|
| D | concatenate_dataset_op_test.cc | 247 TF_EXPECT_OK(VerifyShapesCompatible(dataset_->output_shapes(), in TEST_P()
291 TF_EXPECT_OK(VerifyShapesCompatible(iterator_->output_shapes(), in TEST_P()
|
| D | multi_device_iterator_ops.cc | 75 VerifyShapesCompatible(output_shapes_, iterator->output_shapes())); in Init()
470 VerifyShapesCompatible(output_shapes_, resource->output_shapes())); in VerifyResource()
646 OP_REQUIRES_OK(ctx, VerifyShapesCompatible(output_shapes_, in Compute()
|
| D | zip_dataset_op_test.cc | 229 TF_EXPECT_OK(VerifyShapesCompatible(dataset_->output_shapes(), in TEST_P()
291 TF_EXPECT_OK(VerifyShapesCompatible(iterator_->output_shapes(), in TEST_P()
|
| D | iterator_ops.cc | 176 TF_RETURN_IF_ERROR(VerifyShapesCompatible( in Restore()
249 VerifyShapesCompatible(output_shapes_, iterator->output_shapes())); in SetIteratorFromDataset()
472 VerifyShapesCompatible(output_shapes_, resource->output_shapes())); in VerifyResource()
863 VerifyShapesCompatible(output_shapes_, (*iterator)->output_shapes())); in TryInit()
1111 VerifyShapesCompatible(output_shapes_, in Compute()
|
| D | tensor_dataset_op.cc | 42 OP_REQUIRES_OK(ctx, VerifyShapesCompatible((*output)->output_shapes(), in MakeDataset()
|
| D | tensor_slice_dataset_op.cc | 59 OP_REQUIRES_OK(ctx, VerifyShapesCompatible((*output)->output_shapes(), in MakeDataset()
|
| D | sparse_tensor_slice_dataset_op_test.cc | 263 TF_EXPECT_OK(VerifyShapesCompatible(dataset->output_shapes(), in TEST_F()
400 TF_EXPECT_OK(VerifyShapesCompatible(iterator->output_shapes(), in TEST_F()
|
| D | repeat_dataset_op_test.cc | 288 TF_EXPECT_OK(VerifyShapesCompatible(repeat_dataset->output_shapes(), in TEST_P()
427 TF_EXPECT_OK(VerifyShapesCompatible(iterator->output_shapes(), in TEST_P()
|
| D | take_dataset_op_test.cc | 295 TF_EXPECT_OK(VerifyShapesCompatible(take_dataset->output_shapes(), in TEST_P()
436 TF_EXPECT_OK(VerifyShapesCompatible(iterator->output_shapes(), in TEST_P()
|
| D | dataset_utils.cc | 126 Status VerifyShapesCompatible(const std::vector<PartialTensorShape>& expected, in VerifyShapesCompatible() function
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | indexed_dataset_op.cc | 138 VerifyShapesCompatible(output_shapes_, dataset->output_shapes())); in set_materialized_dataset()
314 VerifyShapesCompatible(output_shapes_, resource->output_shapes())); in VerifyResource()
|