Skip to content

Commit 27774ee

Browse files
feng_shuaifengkuangxiaxia
feng_shuai
andauthored
roi_align fix (#38788) (#38869)
Co-authored-by: fengkuangxiaxia <fengkuangxiaxia21@gmail.com>
1 parent f2a3405 commit 27774ee

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

paddle/fluid/inference/tensorrt/op_teller.cc

+4-1
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,9 @@
1313
// limitations under the License.
1414

1515
#include "paddle/fluid/inference/tensorrt/op_teller.h"
16+
1617
#include <bitset>
18+
1719
#include "paddle/fluid/framework/block_desc.h"
1820
#include "paddle/fluid/framework/data_layout.h"
1921

@@ -1207,7 +1209,8 @@ bool OpTeller::Tell(const framework::ir::Node* node, bool use_no_calib_int8,
12071209
return false;
12081210
}
12091211
std::vector<std::string> attrs{"pooled_height", "pooled_width",
1210-
"spatial_scale", "sampling_ratio"};
1212+
"spatial_scale", "sampling_ratio",
1213+
"aligned"};
12111214
for (auto const attr : attrs) {
12121215
if (!desc.HasAttr(attr)) return false;
12131216
}

0 commit comments

Comments
 (0)