diff --git a/paddle/fluid/inference/tensorrt/op_teller.cc b/paddle/fluid/inference/tensorrt/op_teller.cc index a310ce81a8bdf8e0e5c8ee2c0f28643afd902c25..2899e1da962c2db644aa7e37a745ada34116538f 100644 --- a/paddle/fluid/inference/tensorrt/op_teller.cc +++ b/paddle/fluid/inference/tensorrt/op_teller.cc @@ -13,7 +13,9 @@ // limitations under the License. #include "paddle/fluid/inference/tensorrt/op_teller.h" + #include + #include "paddle/fluid/framework/block_desc.h" #include "paddle/fluid/framework/data_layout.h" @@ -1207,7 +1209,8 @@ bool OpTeller::Tell(const framework::ir::Node* node, bool use_no_calib_int8, return false; } std::vector attrs{"pooled_height", "pooled_width", - "spatial_scale", "sampling_ratio"}; + "spatial_scale", "sampling_ratio", + "aligned"}; for (auto const attr : attrs) { if (!desc.HasAttr(attr)) return false; }