code: def test_vm_shape_of():
      x = relay.var(
          "x",
          shape=(relay.Any(), relay.Any(), relay.Any()),
          dtype="float32",
      )
      relu_x = relay.nn.relu(x)
      data = np.random.uniform(size=(2, 3, 4)).astype("float32")
      args = [data]
  
      newshape_var = relay.var("newshape", shape=(2,), dtype="int64")
      args.append(np.array((1, -1), dtype="int64"))
      main = relay.Function(
          [x, newshape_var],
          relay.reshape(relu_x, newshape=newshape_var),
      )
  
      res = get_serialized_output(main, *args).numpy()
      tvm.testing.assert_allclose(res.flatten(), data.flatten())
  
imports: import pytest
  import numpy
  import tvm
  from tvm.runtime import vm
  from tvm.relay import vm
  from tvm import relay
  from tvm.relay.scope_builder import ScopeBuilder
  from tvm.relay import transform
  from tvm.relay.prelude import Prelude
  from tvm.contrib import utils
  from tvm.relay import testing
label_desc: Use image node_id 15 for calling a global function with example usage: test_vm_shape_of() without return types