Fix parse mean

af29fc68 · Paul · 4ae58245 · af29fc68
Commit af29fc68 authored May 24, 2019 by Paul
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

src/tf/tf.cpp src/tf/tf.cpp +1 -1

No files found.
--- a/src/tf/tf.cpp
+++ b/src/tf/tf.cpp
@@ -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)