Commit 11f33609 authored by turneram's avatar turneram
Browse files

Fix scale and bias ops

parent e47c0140
...@@ -31,9 +31,9 @@ struct parse_layernorm : op_parser<parse_layernorm> ...@@ -31,9 +31,9 @@ struct parse_layernorm : op_parser<parse_layernorm>
make_op("layernorm", {{"epsilon", epsilon}, {"axis", axis}}), args.front()); make_op("layernorm", {{"epsilon", epsilon}, {"axis", axis}}), args.front());
if(args.size() >= 2) if(args.size() >= 2)
layernorm = info.add_instruction(make_op("mul"), layernorm, args.at(1)); layernorm = info.add_broadcastable_binary_op("mul", layernorm, args.at(1));
if(args.size() == 3) if(args.size() == 3)
layernorm = info.add_instruction(make_op("add"), layernorm, args.at(2)); layernorm = info.add_broadcastable_binary_op("add", layernorm, args.at(2));
return layernorm; return layernorm;
} }
......
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