Commit 32b1fda9 authored by Paul's avatar Paul
Browse files

Formatting

parent de13ac71
...@@ -93,7 +93,7 @@ std::vector<float> cpu() ...@@ -93,7 +93,7 @@ std::vector<float> cpu()
auto x = get_tensor_argument_cpu({rtg::shape::float_type, {4, 3, 3, 3}}); auto x = get_tensor_argument_cpu({rtg::shape::float_type, {4, 3, 3, 3}});
auto w = get_tensor_argument_cpu({rtg::shape::float_type, {4, 3, 3, 3}}); auto w = get_tensor_argument_cpu({rtg::shape::float_type, {4, 3, 3, 3}});
p.compile(rtg::cpu::cpu_target{}); p.compile(rtg::cpu::cpu_target{});
auto r = p.eval({{"x", x}, {"w", w}}); auto r = p.eval({{"x", x}, {"w", w}});
auto output = r.get<float>(); auto output = r.get<float>();
result.assign(output.begin(), output.end()); result.assign(output.begin(), output.end());
return result; 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