code: def test_depth_to_space():
      for layout in ["NCHW", "NHWC"]:
          for mode in ["DCR", "CDR"]:
              # Simplest possible case
              verify_depth_to_space(
                  2, 1, 4, 1, 1, layout=layout, mode=mode
              )
              # Average input size
              verify_depth_to_space(
                  2, 1, 32, 32, 32, layout=layout, mode=mode
              )
              # Large block size
              verify_depth_to_space(
                  8, 1, 256, 32, 32, layout=layout, mode=mode
              )
              # Large batch size
              verify_depth_to_space(
                  4, 8, 32, 32, 32, layout=layout, mode=mode
              )
              # Large input size
              verify_depth_to_space(
                  4, 8, 32, 128, 128, layout=layout, mode=mode
              )
  
file_path: tvm/tests/python/topi/test_topi_depth_to_space.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_depth_to_space() without return types