Function TransferFile

tensorflow/core/kernels/file_slice_sendrecv_ops_test.cc:221–256

// Graph Constructor.

static Graph* TransferFile(const std::string& test_type,
                           const int32 slice_size) {
  Graph* g = new Graph(OpRegistry::Global());
  const int64 timeout_ms = 5000;
  std::string recv_dir = "/tmp/FileSliceTransferTestRecv";
  std::string filename = "/tmp/FileSliceTransferTestSend/send_" + test_type;
  std::string contents = \
    "The quick brown fox jumps over the lazy dog."; // 44 chars.

  // send filename node.
  Tensor filename_t(DT_STRING, TensorShape({}));
  filename_t.scalar<tstring>().setConstant(filename);
  Node* filename_n = test::graph::Constant(g, filename_t);

  // contents node.
  Tensor contents_t(DT_STRING, TensorShape({}));
  contents_t.scalar<tstring>().setConstant(contents);
  Node* contents_n = test::graph::Constant(g, contents_t);

  Node* write_file_n = WriteFile(g, filename_n, contents_n);
  Node* send_n = \
    FileSliceSend(g, filename_n, test_type, "/cpu:0", 1, "/cpu:0", slice_size);
  g->AddControlEdge(write_file_n, send_n);

  Node* recv_n = FileSliceRecv(g, test_type, "/cpu:0", 1, "/cpu:0", recv_dir,
                               slice_size, timeout_ms);
  Node* read_file_n = ReadFile(g, recv_n);
  Node* equal_n = Equal(g, contents_n, read_file_n);

  std::vector<NodeBuilder::NodeOut> data_out;
  data_out.emplace_back(contents_n, 0);
  data_out.emplace_back(read_file_n, 0);
  Assert(g, equal_n, data_out);

  return g;
}

static Graph* FileSliceSendTransferFileToSliceRecv(const std::string& test_type,
                                                   const int32 slice_size) {