code: def generated_func(shape: tuple, dtype: str, axis_separators: list):
      """Generate element wise function."""
      dim0, dim1 = shape
  
      @T.prim_func
      def elwise(a: T.handle, b: T.handle):
          a_buffer = T.match_buffer(
              a, shape, dtype=dtype, axis_separators=axis_separators
          )
          b_buffer = T.match_buffer(
              b, shape, dtype=dtype, axis_separators=axis_separators
          )
  
          for i, j in T.grid(dim0, dim1):
              with T.block("compute"):
                  b_buffer[i, j] = a_buffer[i, j] * T.cast(
                      2, dtype=dtype
                  )
  
      return elwise
  
file_path: tvm/tests/python/contrib/test_hexagon/test_memory_alloc.py
imports: import numpy
  import tvm
  from tvm.script import tir
  from tvm.contrib.hexagon import allocate_hexagon_array
  from .infrastructure import get_hexagon_target
label_desc: Use image node_id 1 for calling a global function with example usage: generated_func(shape, dtype, axis_separators) and returns: elwise