test added

apache · Mar 21, 2019 · b42adbb · b42adbb
1 parent 862c348
commit b42adbb
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/tests/python/frontend/mxnet/test_forward.py b/tests/python/frontend/mxnet/test_forward.py
@@ -396,6 +396,12 @@ def verify(shape):
     verify((3, 4, 5, 6))
 
 
+def test_forward_bilinear_resize():
+    data = mx.sym.var('data')
+    mx_sym = mx.sym.contrib.BilinearResize2D(data, height=5, width=10)
+    verify_mxnet_frontend_impl(mx_sym, (1, 2, 3, 4), (1, 2, 5, 10))
+
+
 if __name__ == '__main__':
     test_forward_mlp()
     test_forward_vgg()
@@ -426,3 +432,4 @@ def verify(shape):
     test_forward_slice_axis()
     test_forward_l2_normalize()
     test_forward_shape_array()
+    test_forward_bilinear_resize()