Method Load

oneflow/api/cpp/framework/graph.cpp:232–238

void Graph::set_batch_size(int batch_size) { graph_->set_batch_size(batch_size); }

Graph Graph::Load(const std::string& model_path, const Device& device) {
#ifdef __linux__
  LoadOneEmbedding(model_path, device);
#endif  // __linux__
  Graph graph(model_path, device);
  return graph;
}

Graph::GraphImpl::GraphImpl(const std::string& model_path, const Device& device)
    : model_path_(model_path), device_(device) {