Lines Matching defs:dsl
15 from mlir.dialects.linalg.opdsl import lang as dsl
22 @dsl.linalg_structured_op
24 A=dsl.TensorDef(dsl.T, dsl.S.M, dsl.S.K),
25 B=dsl.TensorDef(dsl.T, dsl.S.K, dsl.S.N),
26 S=dsl.TensorDef(dsl.T, dsl.S.M, dsl.S.N),
27 C=dsl.TensorDef(dsl.T, dsl.S.M, dsl.S.N, output=True),
29 C[dsl.D.m, dsl.D.n] += (
30 S[dsl.D.m, dsl.D.n] * A[dsl.D.m, dsl.D.k] * B[dsl.D.k, dsl.D.n]