porting new upsample test case from nnvm to relay (apache#3115)

neo-ai · May 13, 2019 · b36aeae · b36aeae
1 parent 26e788b
commit b36aeae
Showing 1 changed file with 30 additions and 0 deletions.
diff --git a/tests/python/frontend/onnx/test_forward.py b/tests/python/frontend/onnx/test_forward.py
@@ -427,9 +427,39 @@ def _test_upsample_bilinear():
         tvm_out = get_tvm_output(model, in_array, target, ctx, out_shape, 'float32')
         tvm.testing.assert_allclose(out_array, tvm_out, rtol=1e-5, atol=1e-5)
 
+def _test_upsample_bilinear_opset9():
+    scale = 2
+    in_shape = (1, 1, 3, 3)
+    out_shape = (1, 1, 3*scale, 3*scale)
+    y = helper.make_node("Upsample", ['in','scales'], ['out'], mode='linear')
+    scales=[1.0, 1.0, 2.0, 2.0]
+    in_array = np.random.uniform(size=in_shape).astype(np.float32)
+    out_array = topi.testing.bilinear_resize_python(in_array, (3*scale, 3*scale), "NCHW")
+
+    ref_array = np.array(scales)
+    ref_node = helper.make_node('Constant',
+                                 inputs=[],
+                                 outputs=['scales'],
+                                 value=onnx.helper.make_tensor(name = 'const_tensor',
+                                                               data_type = TensorProto.FLOAT,
+                                                               dims = ref_array.shape,
+                                                               vals = ref_array.flatten().astype(float)))
+
+    graph = helper.make_graph([ref_node, y],
+                              'upsample_bilinear_opset9_test',
+                              inputs = [helper.make_tensor_value_info("in", TensorProto.FLOAT, list(in_shape))],
+                              outputs = [helper.make_tensor_value_info("out", TensorProto.FLOAT, list(out_shape))])
+
+    model = helper.make_model(graph, producer_name='upsample_bilinear_opset9_test')
+
+    for target, ctx in ctx_list():
+        tvm_out = get_tvm_output(model, in_array, target, ctx, out_shape, 'float32')
+        tvm.testing.assert_allclose(out_array, tvm_out, rtol=1e-5, atol=1e-5)
+
 def test_upsample():
     _test_upsample_nearest()
     _test_upsample_bilinear()
+    _test_upsample_bilinear_opset9()
 
 def _test_softmax(inshape, axis):
     opname = 'Softmax'