tvm_tirx.backend.metal.op.cooperative_tensor_store

tvm_tirx.backend.metal.op.cooperative_tensor_store(d, index, ptr, stride, rows, cols, transpose_matrix=False, mma_M=0, mma_N=0, mma_K=0, operand_role=0, *, ty=None, loc=UNKNOWN_LOC)

Parameters:

• d (Expr)

• index (Expr)

• ptr (Expr)

• stride (Expr)

• rows (int)

• cols (int)

• transpose_matrix (bool)

• mma_M (int)

• mma_N (int)

• mma_K (int)

• operand_role (int)

• loc (Location)