Method AddTensorHeader

tensorflow/core/util/tensor_bundle/tensor_bundle.cc:503–519

}

Status BundleWriter::AddTensorHeader(StringPiece key, DataType dtype) {
  if (!status_.ok()) return status_;
  CHECK_NE(key, kHeaderEntryKey);
  const string key_string(key);
  if (entries_.find(key_string) != entries_.end()) {
    status_ = errors::InvalidArgument("Adding duplicate key: ", key);
    return status_;
  }

  entry_seg_ = &entries_[key_string];
  entry_seg_->set_dtype(dtype);
  entry_seg_->set_shard_id(0);
  entry_seg_->set_offset(size_);

  out_->clear_crc32c();
  return status_;
}

Status BundleWriter::AddSliceHeader(
    string tensor_name, const TensorShape& shape, DataType type, bool is_hash,