tvm_tirx.backend.metal.op.cooperative_tensor_multiply_accumulate

tvm_tirx.backend.metal.op.cooperative_tensor_multiply_accumulate(d, index_d, a, index_a, b, index_b, c, index_c, M, N, K, transpose_a=False, transpose_b=False, *, ty=None, loc=UNKNOWN_LOC)

Parameters:

• d (Var)

• index_d (Expr)

• a (Var)

• index_a (Expr)

• b (Var)

• index_b (Expr)

• c (Var)

• index_c (Expr)

• M (int)

• N (int)

• K (int)

• transpose_a (bool)

• transpose_b (bool)

• loc (Location)