code: def test_space_to_depth():
      for layout in ["NCHW", "NHWC"]:
          # Simplest possible case
          verify_space_to_depth(2, 1, 1, 2, 2, layout=layout)
          # Average input size
          verify_space_to_depth(2, 1, 32, 32, 32, layout=layout)
          # Large block size
          verify_space_to_depth(8, 1, 32, 64, 64, layout=layout)
          # Large batch size
          verify_space_to_depth(4, 8, 32, 32, 32, layout=layout)
          # Large input size
          verify_space_to_depth(4, 8, 32, 128, 128, layout=layout)
  
file_path: tvm/tests/python/topi/test_topi_space_to_depth.py
imports: import numpy
  import tvm
  from tvm import te
  from tvm import topi
  import tvm.testing
  import tvm.topi.testing
label_desc: Use image node_id 2 for calling a global function with example usage: test_space_to_depth() without return types