Commit ca4eb13d authored by Khalique's avatar Khalique
Browse files

formatting

parent 5cd9877d
...@@ -98,7 +98,7 @@ struct tf_parser ...@@ -98,7 +98,7 @@ struct tf_parser
{ {
if(is_nhwc) if(is_nhwc)
{ {
l0 = prog.add_instruction(op::transpose{{0,3,1,2}}, args[1]); l0 = prog.add_instruction(op::transpose{{0, 3, 1, 2}}, args[1]);
} }
} }
return add_broadcastable_binary_op(args[0], l0, x); return add_broadcastable_binary_op(args[0], l0, x);
...@@ -162,8 +162,7 @@ struct tf_parser ...@@ -162,8 +162,7 @@ struct tf_parser
{ {
float epsilon = 1e-5f; float epsilon = 1e-5f;
float momentum = 0.9f; float momentum = 0.9f;
op::batch_norm_inference::bn_infer_mode_t bn_mode = op::batch_norm_inference::bn_infer_mode_t bn_mode = op::batch_norm_inference::spatial;
op::batch_norm_inference::spatial;
if(contains(attributes, "epsilon")) if(contains(attributes, "epsilon"))
{ {
epsilon = attributes.at("epsilon").f(); epsilon = attributes.at("epsilon").f();
...@@ -385,7 +384,7 @@ struct tf_parser ...@@ -385,7 +384,7 @@ struct tf_parser
auto l0 = args[0]; auto l0 = args[0];
if(is_nhwc) if(is_nhwc)
{ {
l0 = prog.add_instruction(op::transpose{{0,2,3,1}}, args[0]); l0 = prog.add_instruction(op::transpose{{0, 2, 3, 1}}, args[0]);
} }
return prog.add_instruction(op, l0); return prog.add_instruction(op, l0);
} }
......
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