code: def te_matmul():
      k = te.reduce_axis((0, 128), "k")
      A = te.placeholder((128, 128), name="A")
      B = te.placeholder((128, 128), name="B")
      C = te.compute(
          (128, 128),
          lambda x, y: te.sum(A[x, k] * B[y, k], axis=k),
          name="C",
      )
      return [A, B, C]
  
imports: import numpy
  import tvm
  import tvm.testing
  from tvm import te, tir, topi, relay
  from tvm.script import tir
  import pytest
label_desc: Use image node_id 4 for calling a global function with example usage: te_matmul() and returns: list