code: def test_yolo_reorg_infer_shape():
      def verify_yolo_reorg(shape, stride, out_shape):
          x = relay.var("x", relay.TensorType(shape, "float32"))
          z = relay.vision.yolo_reorg(x, stride=stride)
          zz = run_infer_type(z)
          assert "stride=" in z.astext()
          assert zz.checked_type == relay.ty.TensorType(
              out_shape, "float32"
          )
  
      n, c, h, w = (
          te.size_var("n"),
          te.size_var("c"),
          te.size_var("h"),
          te.size_var("w"),
      )
      idxd = tvm.tir.indexdiv
      verify_yolo_reorg((n, c, 20, 20), 10, (n, c * 10 * 10, 2, 2))
      verify_yolo_reorg(
          (n, c, h, w), 2, (n, c * 2 * 2, idxd(h, 2), idxd(w, 2))
      )
  
imports: import math
  import platform
  import sys
  import numpy
  import pytest
  import tvm
  import tvm.testing
  import tvm.topi.testing
  from tvm import relay, te
  from tvm.relay.testing import run_infer_type
label_desc: Use image node_id 11 for calling a global function with example usage: test_yolo_reorg_infer_shape() without return types