fix reduce_mean issue

PaddlePaddle · Jul 28, 2021 · f495c0e · f495c0e · paddle-bot-old · Jul 29, 2021
1 parent a43d53d
commit f495c0e
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/paddle/fluid/inference/tensorrt/op_teller.cc b/paddle/fluid/inference/tensorrt/op_teller.cc
@@ -719,12 +719,14 @@ bool OpTeller::Tell(const framework::ir::Node* node, bool use_no_calib_int8,
         VLOG(3) << "the " << op_type
                 << " does not have attr (keep_dim or dim or "
                    "reduce_all)";
+        std::cout << "attr " << desc.HasAttr("keep_dim") << " "
+                  << desc.HasAttr("dim") << " " << desc.HasAttr("reduce_all");
         return false;
       }
 
       // The batch size dimension cannot be reduced if it's not dynamic shape.
       if (!with_dynamic_shape) {
-        if (desc.HasAttr("reduce_all")) return false;
+        if (BOOST_GET_CONST(bool, desc.GetAttr("reduce_all"))) return false;
         std::vector<int32_t> dim =
             BOOST_GET_CONST(std::vector<int32_t>, desc.GetAttr("dim"));
         for (auto x : dim) {