From 31cbf63eb1125a01f76d59880ec97a1553b182f0 Mon Sep 17 00:00:00 2001 From: ccrrong <101700995+ccrrong@users.noreply.github.com> Date: Tue, 21 Jun 2022 15:51:35 +0800 Subject: [PATCH] fix scale init error (#43638) --- .../fluid/inference/tensorrt/convert/bilinear_interp_v2_op.cc | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/paddle/fluid/inference/tensorrt/convert/bilinear_interp_v2_op.cc b/paddle/fluid/inference/tensorrt/convert/bilinear_interp_v2_op.cc index f0e56082b8f..e303e198888 100644 --- a/paddle/fluid/inference/tensorrt/convert/bilinear_interp_v2_op.cc +++ b/paddle/fluid/inference/tensorrt/convert/bilinear_interp_v2_op.cc @@ -60,8 +60,8 @@ class BilinearInterpolateV2OpConverter : public OpConverter { } auto in_dim = input->getDimensions(); - float scale_h = 1.f; - float scale_w = 1.f; + float scale_h = -1.f; + float scale_w = -1.f; // Scales Priority: Scale(tensor) > scale(attr) > out_d/out_h/out_w(attr) bool has_scale_input_attr = -- GitLab