diff --git a/tests/python/frontend/mxnet/test_forward.py b/tests/python/frontend/mxnet/test_forward.py index e83f1e569545e..9880bb39a44e9 100644 --- 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()