Function WriteTensor

tensorflow/core/util/tensor_bundle/tensor_bundle.cc:225–233

Serializes the data bytes of the non-string tensor "val". Discards the original content of "bytes_written", and on OK updates it with number of bytes written. REQUIRES: val.dtype() != DT_STRING

// bytes written.
// REQUIRES: val.dtype() != DT_STRING
Status WriteTensor(const Tensor& val, FileOutputBuffer* out,
                   size_t* bytes_written) {
  DCHECK_NE(val.dtype(), DT_STRING);
  DCHECK_NE(val.dtype(), DT_VARIANT);
  *bytes_written = val.TotalBytes();
  char* buf = GetBackingBuffer(val);
  VLOG(1) << "Appending " << *bytes_written << " bytes to file";
  return out->Append(StringPiece(buf, *bytes_written));
}

// Serializes string tensor "val".  "bytes_written" is treated in the same
// fashion as WriteTensor().