Method WriteToPath

tensorflow/compiler/xla/text_literal_writer.cc:32–59

static */

namespace xla {

/* static */ Status TextLiteralWriter::WriteToPath(const Literal& literal,
                                                   absl::string_view path) {
  std::unique_ptr<tensorflow::WritableFile> f;
  auto s = tensorflow::Env::Default()->NewWritableFile(string(path), &f);
  if (!s.ok()) {
    return s;
  }

  s = f->Append(ShapeUtil::HumanString(literal.shape()) + "\n");
  if (!s.ok()) {
    return s;
  }

  Status status;
  tensorflow::WritableFile* f_ptr = f.get();
  literal.EachCellAsString(
      [f_ptr, &status](absl::Span<const int64> indices, const string& value) {
        if (!status.ok()) {
          return;
        }
        string coordinates =
            absl::StrCat("(", absl::StrJoin(indices, ", "), ")");

        status = f_ptr->Append(absl::StrCat(coordinates, ": ", value, "\n"));
      });
  auto ignored = f->Close();
  return status;
}

}  // namespace xla