code: def compute_dot(A, B):
      M = A.shape[0]
      K = A.shape[1]
      N = B.shape[1]
      k = tvm.te.reduce_axis((0, K), "k")
      C = tvm.te.compute(
          (M, N),
          lambda x, y: tvm.tir.sum(A[x, k] * B[k, y], axis=k),
          name="C",
      )
      return C
  
imports: import tvm
  from tvm import autotvm
  from ..None import defop, AllTypes
  from ..None import assign_by_req, reduce_axes
label_desc: Use image node_id 1 for calling a global function with example usage: compute_dot(A, B) and returns: C