[TOPI] Fix uint8 resize_bilinear issue. (apache#2490)

Signed-off-by: Zhebin Jin <[email protected]>
neo-ai · Feb 20, 2019 · 1ad84f8 · 1ad84f8
1 parent d42d218
commit 1ad84f8
Showing 1 changed file with 8 additions and 8 deletions.
diff --git a/topi/include/topi/image/resize.h b/topi/include/topi/image/resize.h
@@ -54,10 +54,10 @@ inline Expr bilinear_sample_nchw(const Tensor& input, const Array<Expr>& indices
   auto C = input(indices[0], indices[1], y1, x0);
   auto D = input(indices[0], indices[1], y1, x1);
 
-  auto top = A + (B - A) * x_lerp;
-  auto bottom = C + (D - C) * x_lerp;
-
-  return (top + (bottom - top) * y_lerp);
+  return A * ( 1 - x_lerp) * ( 1 - y_lerp) +
+         B * x_lerp * (1 - y_lerp) +
+         C * (1 - x_lerp) * y_lerp +
+         D * x_lerp * y_lerp;
 }
 
 /*!
@@ -231,10 +231,10 @@ inline Tensor resize_bilinear_nhwc(const Tensor& input,
     auto C = input(indices[0], y1, x0, indices[3]);
     auto D = input(indices[0], y1, x1, indices[3]);
 
-    auto top = A + (B - A) * x_lerp;
-    auto bottom = C + (D - C) * x_lerp;
-
-    return  (top + (bottom - top) * y_lerp);
+    return A * ( 1 - x_lerp) * ( 1 - y_lerp) +
+           B * x_lerp * (1 - y_lerp) +
+           C * (1 - x_lerp) * y_lerp +
+           D * x_lerp * y_lerp;
     }, name, tag);
 }