Commit 8c2b3895 authored by Khalique's avatar Khalique
Browse files

formatting

parent 8b023ae3
...@@ -426,13 +426,13 @@ struct onnx_parser ...@@ -426,13 +426,13 @@ struct onnx_parser
{ {
if(beta != 0.f) if(beta != 0.f)
{ {
auto l3 = prog.add_instruction(op::dot{alpha}, l1, l2); auto l3 = prog.add_instruction(op::dot{alpha}, l1, l2);
auto l4 = args[2]; auto l4 = args[2];
if(beta == 1.f) if(beta == 1.f)
{ {
auto beta_val = prog.add_literal(beta); auto beta_val = prog.add_literal(beta);
auto l5 = prog.add_instruction(op::scalar{args[2]->get_shape()}, beta_val); auto l5 = prog.add_instruction(op::scalar{args[2]->get_shape()}, beta_val);
l4 = prog.add_instruction(op::mul{}, args[2], l5); l4 = prog.add_instruction(op::mul{}, args[2], l5);
} }
return add_broadcastable_binary_op(l3, l4, op::add{}); return add_broadcastable_binary_op(l3, l4, op::add{});
} }
......
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