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 C=dsl.TensorDef(dsl.T, dsl.S.M, dsl.S.N, output=True),
28 C[dsl.D.m, dsl.D.n] += A[dsl.D.m, dsl.D.k] * B[dsl.D.k, dsl.D.n]