Commit 708c0401 authored by Khalique's avatar Khalique
Browse files

formatting

parent 0420b20d
......@@ -589,8 +589,7 @@ struct tf_parser
auto l0 = prog.add_instruction(op::reduce_mean{axes_int64}, args.front());
if(keep_dims)
return l0;
return prog.add_instruction(
op::squeeze{axes_int64}, l0);
return prog.add_instruction(op::squeeze{axes_int64}, l0);
// }
// MIGRAPHX_THROW("MIGraphX does not support mean outside of GlobalAvgPool transformation");
}
......
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