Searched refs:VerifyShapesCompatible (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | iterator_ops.cc | 261 VerifyShapesCompatible(output_shapes_, iterator->output_shapes())); in SetIteratorFromDataset()
525 VerifyShapesCompatible(output_shapes_, resource->output_shapes())); in VerifyResource()
659 TF_RETURN_IF_ERROR(VerifyShapesCompatible(output_shapes_, components)); in DoCompute()
758 TF_RETURN_IF_ERROR(VerifyShapesCompatible(output_shapes_, state)); in DoCompute()
880 VerifyShapesCompatible(output_shapes_, (*iterator)->output_shapes())); in TryInit()
997 TF_RETURN_IF_ERROR(VerifyShapesCompatible(output_shapes_, components)); in DoCompute()
1108 VerifyShapesCompatible(output_shapes_, in Compute()
|
| D | dataset_utils.h | 96 Status VerifyShapesCompatible(const std::vector<PartialTensorShape>& expected,
99 Status VerifyShapesCompatible(const std::vector<PartialTensorShape>& expected,
|
| D | tensor_dataset_op.cc | 163 ctx, VerifyShapesCompatible((*output)->output_shapes(), output_shapes_)); in MakeDataset()
|
| D | multi_device_iterator_ops.cc | 84 VerifyShapesCompatible(output_shapes_, iterator->output_shapes())); in Init()
493 VerifyShapesCompatible(output_shapes_, resource->output_shapes())); in VerifyResource()
727 OP_REQUIRES_OK(ctx, VerifyShapesCompatible(output_shapes_, in Compute()
|
| D | tensor_slice_dataset_op.cc | 205 ctx, VerifyShapesCompatible((*output)->output_shapes(), output_shapes_)); in MakeDataset()
|
| D | dataset_utils.cc | 152 Status VerifyShapesCompatible(const std::vector<PartialTensorShape>& expected, in VerifyShapesCompatible() function
166 Status VerifyShapesCompatible(const std::vector<PartialTensorShape>& expected, in VerifyShapesCompatible() function
|
| D | dataset_test_base.cc | 693 TF_EXPECT_OK(VerifyShapesCompatible(dataset_->output_shapes(), in CheckDatasetOutputShapes()
712 TF_EXPECT_OK(VerifyShapesCompatible(iterator_->output_shapes(), in CheckIteratorOutputShapes()
|