Method AddSliceHeader

tensorflow/core/util/tensor_bundle/tensor_bundle.cc:521–538

}

Status BundleWriter::AddSliceHeader(
    string tensor_name, const TensorShape& shape, DataType type, bool is_hash,
    TensorSliceProto** proto) {
  if (!status_.ok()) return status_;
  BundleEntryProto* full_entry = &entries_[tensor_name];
  if (full_entry->dtype() != DT_INVALID) {
    CHECK_EQ(full_entry->dtype(), type);
  }
  if (full_entry->has_shape()) {
    CHECK(TensorShape(full_entry->shape()) == shape);
  }

  full_entry->set_is_hash_table(is_hash);
  full_entry->set_dtype(type);
  shape.AsProto(full_entry->mutable_shape());
  *proto = full_entry->add_slices();
  return Status::OK();
}

Status BundleWriter::AddTensorHeader(StringPiece key, DataType dtype, TensorShape shape) {
  if (!status_.ok()) return status_;