Commit de13ac71 authored by Paul's avatar Paul
Browse files

Use get method

parent 91bff175
......@@ -94,7 +94,8 @@ std::vector<float> cpu()
auto w = get_tensor_argument_cpu({rtg::shape::float_type, {4, 3, 3, 3}});
p.compile(rtg::cpu::cpu_target{});
auto r = p.eval({{"x", x}, {"w", w}});
r.visit([&](auto output) { result.assign(output.begin(), output.end()); });
auto output = r.get<float>();
result.assign(output.begin(), output.end());
return result;
}
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment