Commit af29fc68 authored by Paul's avatar Paul
Browse files

Fix parse mean

parent 4ae58245
......@@ -446,7 +446,7 @@ struct tf_parser
{
auto axes = args[1]->eval().get<int32_t>().to_vector();
bool keep_dims = attributes.at("keep_dims").b();
std::vector<int32_t> hw_axes{2, 3};
std::vector<int32_t> hw_axes{1, 2};
// check if conditions for GlobalAvgPool are met
auto lens = args[0]->get_shape().lens();
if(axes == hw_axes and lens.size() == 4)
......
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