code: def get_after_storage_legalize():
      @tvm.script.ir_module
      class After:
          @T.prim_func
          def main(
              Aptr: T.handle("uint16"),
              Bptr: T.handle("uint16"),
              Dptr: T.handle("uint16"),
          ):
              T.func_attr({"global_symbol": "main"})
              A = T.decl_buffer((100,), "uint16", data=Aptr)
              B = T.decl_buffer((100,), "uint16", data=Bptr)
              D = T.decl_buffer((100,), "uint16", data=Dptr)
              C = T.decl_buffer((100,), "float32")
              for i in T.grid(100):
                  C[i] = u16tof32(A[i]) + u16tof32(B[i])
                  D[i] = f32tou16(T.exp(C[i]))
  
      return After
  
file_path: tvm/tests/python/tir-transform/test_tir_transform_bf16_legalize.py
imports: import tvm
  import tvm.script
  from tvm.script import tir
label_desc: Use image node_id 7 for calling a global function with example usage: get_after_storage_legalize() and returns: After