Method GetTensorInfo

tensorflow/core/util/tensor_bundle/tensor_bundle.cc:1165–1175

}

Status BundleReader::GetTensorInfo(
    StringPiece key, int64* size,
    std::unique_ptr<RandomAccessFile>* file, int64* offset) {
  BundleEntryProto entry;
  TF_RETURN_IF_ERROR(GetBundleEntryProto(key, &entry));
  TF_RETURN_IF_ERROR(env_->NewRandomAccessFile(
        DataFilename(prefix_, entry.shard_id(), num_shards_), file));
  *size = entry.size();
  *offset = entry.offset();
  return Status::OK();
}

Status BundleReader::ReadCurrent(Tensor* val) {
  CHECK(val != nullptr);