diff options
author | 2018-09-23 08:53:45 -0700 | |
---|---|---|
committer | 2018-09-23 08:58:05 -0700 | |
commit | f1237459efb3a5578885b03d5b33c3fed350c348 (patch) | |
tree | 128b49a2112e773640a06b3b0578afa8cddd81f0 /tensorflow/core/kernels/data/iterator_ops.cc | |
parent | 646b3c237deaddddd087d39ab57130b08375c4c7 (diff) |
Moving MultiDeviceIterator from contrib to core.
PiperOrigin-RevId: 214173896
Diffstat (limited to 'tensorflow/core/kernels/data/iterator_ops.cc')
-rw-r--r-- | tensorflow/core/kernels/data/iterator_ops.cc | 37 |
1 files changed, 0 insertions, 37 deletions
diff --git a/tensorflow/core/kernels/data/iterator_ops.cc b/tensorflow/core/kernels/data/iterator_ops.cc index 30c6585ba2..c0bc507ec0 100644 --- a/tensorflow/core/kernels/data/iterator_ops.cc +++ b/tensorflow/core/kernels/data/iterator_ops.cc @@ -44,43 +44,6 @@ namespace { const char kIteratorVariantTypeName[] = "tensorflow::Iterator"; -Status VerifyTypesMatch(const DataTypeVector& expected, - const DataTypeVector& received) { - if (expected.size() != received.size()) { - return errors::InvalidArgument( - "Number of components does not match: expected ", expected.size(), - " types but got ", received.size(), "."); - } - for (size_t i = 0; i < expected.size(); ++i) { - if (expected[i] != received[i]) { - return errors::InvalidArgument("Data type mismatch at component ", i, - ": expected ", DataTypeString(expected[i]), - " but got ", DataTypeString(received[i]), - "."); - } - } - return Status::OK(); -} - -Status VerifyShapesCompatible(const std::vector<PartialTensorShape>& expected, - const std::vector<PartialTensorShape>& received) { - if (expected.size() != received.size()) { - return errors::InvalidArgument( - "Number of components does not match: expected ", expected.size(), - " shapes but got ", received.size(), "."); - } - for (size_t i = 0; i < expected.size(); ++i) { - if (!expected[i].IsCompatibleWith(received[i])) { - return errors::InvalidArgument("Incompatible shapes at component ", i, - ": expected ", expected[i].DebugString(), - " but got ", received[i].DebugString(), - "."); - } - } - - return Status::OK(); -} - } // namespace class IteratorResource : public ResourceBase { |