test=develop

b3602sss · b3602sss · commit cc93a67b5eba · 2021-07-07T10:39:17.000Z
diff --git a/paddle/fluid/inference/tensorrt/op_teller.cc b/paddle/fluid/inference/tensorrt/op_teller.cc
@@ -320,6 +320,7 @@ bool OpTeller::Tell(const framework::ir::Node* node, bool use_no_calib_int8,
       for (auto& input : inputs) {
         if (input == "Axis" && desc.Input("Axis").size() > 0) return false;
       }
+
       // current not support axis from input, use default 0
       if (desc.GetAttrIfExists<int>("axis")) return false;
     }

Original file line number	Diff line number	Diff line change
`@@ -320,6 +320,7 @@ bool OpTeller::Tell(const framework::ir::Node* node, bool use_no_calib_int8,`
`320`	`320`	`for (auto& input : inputs) {`
`321`	`321`	`if (input == "Axis" && desc.Input("Axis").size() > 0) return false;`
`322`	`322`	`}`
	`323`	`+`
`323`	`324`	`// current not support axis from input, use default 0`
`324`	`325`	`if (desc.GetAttrIfExists<int>("axis")) return false;`
`325`	`326`	`}`