Commit 4c7956f7 authored by turneram's avatar turneram
Browse files

Formatting

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