Method ReadFileSlice

tensorflow/core/kernels/file_slice_sendrecv_ops.cc:223–237

}

Status FileSliceSendOp::ReadFileSlice(
                          const std::unique_ptr<RandomAccessFile>& file,
                          const uint64 pos, const uint64 offset,
                          Tensor& data_t) {
  string* data_s = data_t.scalar<tstring>().data();
  gtl::STLStringResizeUninitialized(data_s, offset);
  char* data_p = gtl::string_as_array(data_s);
  StringPiece result;
  TF_RETURN_IF_ERROR(file->Read(pos, offset, &result, data_p));
  if (result.data() != data_p) {
    memmove(data_p, result.data(), result.size());
  }

  return Status::OK();
}

REGISTER_KERNEL_BUILDER(Name("_FileSliceSend").Device(DEVICE_CPU),
                        FileSliceSendOp);