MCPcopy Create free account
hub / github.com/Oneflow-Inc/oneflow / Run

Method Run

oneflow/api/cpp/framework/graph.cpp:316–326  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

314}
315
316of::Maybe<std::vector<Tensor>> Graph::GraphImpl::Run(const std::vector<Tensor>& inputs) const {
317 const auto input_tensor_tuple = std::make_shared<of::one::TensorTuple>();
318 for (const auto& tensor : inputs) { input_tensor_tuple->emplace_back(tensor.tensor_); }
319
320 JUST(of::RunLazyNNGraph(*input_tensor_tuple, *output_tensor_tuple_, graph_));
321 JUST(of::SoftSyncNNGraphBuffers(*output_tensor_tuple_, graph_));
322
323 std::vector<Tensor> outputs;
324 for (const auto& tensor : *output_tensor_tuple_) { outputs.emplace_back(Tensor(tensor)); }
325 return outputs;
326}
327
328of::Maybe<void> Graph::GraphImpl::AddOp(of::OperatorConf op_conf) {
329 {

Callers

nothing calls this directly

Calls 4

RunLazyNNGraphFunction · 0.85
SoftSyncNNGraphBuffersFunction · 0.85
TensorClass · 0.70
emplace_backMethod · 0.45

Tested by

no test coverage detected