fix scale init error (#43638)

31cbf63e · ccrrong · GitHub · 81beec1c · 31cbf63e
隐藏空白更改
内联并排

Showing with 2 addition and 2 deletion

paddle/fluid/inference/tensorrt/convert/bilinear_interp_v2_op.cc ...fluid/inference/tensorrt/convert/bilinear_interp_v2_op.cc +2 -2

未找到文件。
--- 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 =