code: def get_shape(
      im_height,
      im_width,
      in_filter,
      out_filter,
      k_h,
      k_w,
      hpad,
      wpad,
      hstride,
      wstride,
      out_dtype,
  ):
      """
      Finds out the shape of all data structures
      """
      data_shape = (
          1,
          in_filter // NUM_VEC_LANES,
          im_height,
          im_width,
          NUM_VEC_LANES,
      )
  
      if out_dtype == "int32" or out_dtype == "uint32":
          kernel_shape = (
              out_filter // NUM_VEC_LANES,
              in_filter // NUM_VEC_LANES,
              k_h,
              k_w,
              NUM_VEC_LANES // 4,
              NUM_VEC_LANES,
              4,
          )
      elif out_dtype == "float32":
          kernel_shape = (
              out_filter // NUM_VEC_LANES,
              in_filter // NUM_VEC_LANES,
              k_h,
              k_w,
              NUM_VEC_LANES,
              NUM_VEC_LANES,
          )
      out_height = (im_height + 2 * hpad - k_h) // hstride + 1
      out_width = (im_width + 2 * wpad - k_w) // wstride + 1
      o_shape = (
          1,
          out_filter // NUM_VEC_LANES,
          out_height,
          out_width,
          NUM_VEC_LANES,
      )
      return (data_shape, kernel_shape, o_shape)
  
file_path: tvm/apps/topi_recipe/conv/test_conv_int8_arm.py
imports: import sys
  import logging
  import numpy
  import tvm
  from tvm import te
  from tvm import topi
label_desc: Use image node_id 1 for calling a global function with example usage: get_shape(im_height, im_width, in_filter, out_filter, k_h, k_w, hpad, wpad, hstride, wstride, out_dtype) and returns: data_shape,  kernel_shape,  o_shape