code: def _get_model(shape, block, dtype, layout):
      a = relay.var("a", shape=shape, dtype=dtype)
      depth = relay.nn.depth_to_space(
          a, layout=layout, block_size=block
      )
      return depth
  
file_path: tvm/tests/python/contrib/test_ethosn/test_depth_to_space.py
imports: import pytest
  import numpy
  import tvm
  from tvm import relay
  from tvm.testing import requires_ethosn
  from .None import infrastructure
label_desc: Use image node_id 1 for calling a global function with example usage: _get_model(shape, block, dtype, layout) and returns: depth