Home
last modified time | relevance | path

Searched refs:TensorDef (Results 1 – 17 of 17) sorted by relevance

/llvm-project-15.0.7/mlir/python/mlir/dialects/linalg/opdsl/ops/
H A Dcore_named_ops.py10 def copy(I=TensorDef(T1),
11 O=TensorDef(U, output=True),
22 def elemwise_unary(I=TensorDef(T1),
35 def elemwise_binary(lhs=TensorDef(T1),
36 rhs=TensorDef(T2),
49 def matmul(A=TensorDef(T1, S.M, S.K),
50 B=TensorDef(T2, S.K, S.N),
156 y=TensorDef(T2, S.N),
169 def vecmat(y=TensorDef(T1, S.M),
198 def dot(A=TensorDef(T1, S.M), B=TensorDef(T2, S.M), C=TensorDef(U,
[all …]
/llvm-project-15.0.7/mlir/test/python/dialects/linalg/opdsl/
H A Dshape_maps_iteration.py23 A=TensorDef(T, S.M, S.K),
24 B=TensorDef(T, S.K, S.N),
25 C=TensorDef(U, S.M, S.N, output=True)):
45 def dot(A=TensorDef(T, S.M), B=TensorDef(T, S.M), C=TensorDef(U, output=True)):
65 I=TensorDef(T, S.I),
66 K=TensorDef(T, S.K, index_dims=[D.k]),
67 O=TensorDef(U, S.O, output=True)):
H A Darguments.py32 A=TensorDef(T, S.M, S.K),
33 B=TensorDef(T, S.K, S.N),
34 C=TensorDef(U, S.M, S.N, output=True),
49 def fill(value=ScalarDef(T), O=TensorDef(T, S.M, S.K, output=True)):
72 I=TensorDef(T, S.IH, S.IW),
73 O=TensorDef(T, S.OH, S.OW, output=True),
H A Dassignments.py34 A=TensorDef(T, S.M, S.K),
35 B=TensorDef(T, S.K, S.N),
36 C=TensorDef(U, S.M, S.N, output=True),
82 O=TensorDef(T, S.M, S.K, output=True),
102 def indices(O=TensorDef(T, S.M, S.K, output=True)):
113 def fill(value=ScalarDef(T), O=TensorDef(T, S.M, S.K, output=True)):
H A Demit_misc.py19 def test_const(O=TensorDef(F32, S.M, S.N, output=True)):
25 def test_index(O=TensorDef(I32, S.M, S.N, output=True)):
32 I=TensorDef(T),
33 O=TensorDef(U, output=True),
40 def non_default_op_name(I=TensorDef(T, S.N), O=TensorDef(T, S.N, output=True)):
H A Demit_matmul.py16 A=TensorDef(T, S.M, S.K),
17 B=TensorDef(T, S.K, S.N),
18 C=TensorDef(T, S.M, S.N, output=True)):
25 A=TensorDef(T1, S.M, S.K),
26 B=TensorDef(T2, S.K, S.N),
27 C=TensorDef(U, S.M, S.N, output=True),
H A Dmetadata.py14 A=TensorDef(T, S.M, S.K),
15 B=TensorDef(T, S.K, S.N),
16 C=TensorDef(U, S.M, S.N, output=True)):
H A Demit_fill.py15 def fill_poly(value=ScalarDef(T1), O=TensorDef(U, output=True)):
19 def fill_rank_zero_poly(I=TensorDef(T1), O=TensorDef(U, output=True)):
H A Demit_convolution.py16 I=TensorDef(T1, S.N, S.IH, S.IW, S.C),
17 K=TensorDef(T2, S.KH, S.KW, S.C),
18 O=TensorDef(U, S.N, S.OH, S.OW, S.C, output=True),
H A Demit_pooling.py16 I=TensorDef(T1, S.N, S.H, S.W, S.C),
17 K=TensorDef(T2, S.KH, S.KW, index_dims=[D.kh, D.kw]),
18 O=TensorDef(U, S.N, S.OH, S.OW, S.C, output=True),
/llvm-project-15.0.7/mlir/docs/Dialects/Linalg/
H A DOpDSL.md49 def matmul(A=TensorDef(T1, S.M, S.K),
50 B=TensorDef(T2, S.K, S.N),
51 C=TensorDef(U, S.M, S.N, output=True)):
97 I=TensorDef(T, S.M, S.K),
124 def strided_copy(I=TensorDef(T, S.IH, S.IW),
163 I=TensorDef(T1, S.N, S.H, S.W, S.C),
165 O=TensorDef(U, S.N, S.OH, S.OW, S.C, output=True),
239 lhs=TensorDef(T1),
240 rhs=TensorDef(T2),
241 O=TensorDef(U, output=True),
[all …]
/llvm-project-15.0.7/mlir/test/Integration/Dialect/SparseTensor/python/
H A Dtest_SDDMM.py23 A=dsl.TensorDef(dsl.T, dsl.S.M, dsl.S.K),
24 B=dsl.TensorDef(dsl.T, dsl.S.K, dsl.S.N),
25 S=dsl.TensorDef(dsl.T, dsl.S.M, dsl.S.N),
26 C=dsl.TensorDef(dsl.T, dsl.S.M, dsl.S.N, output=True)):
H A Dtest_SpMM.py23 A=dsl.TensorDef(dsl.T, dsl.S.M, dsl.S.K),
24 B=dsl.TensorDef(dsl.T, dsl.S.K, dsl.S.N),
25 C=dsl.TensorDef(dsl.T, dsl.S.M, dsl.S.N, output=True)):
/llvm-project-15.0.7/mlir/benchmark/python/
H A Dbenchmark_sparse.py26 A=dsl.TensorDef(dsl.T, dsl.S.M, dsl.S.K),
27 B=dsl.TensorDef(dsl.T, dsl.S.K, dsl.S.N),
28 C=dsl.TensorDef(dsl.T, dsl.S.M, dsl.S.N, output=True)
/llvm-project-15.0.7/mlir/test/mlir-linalg-ods-gen/
H A Dtest-linalg-ods-yaml-gen.yaml5 # def test1(O=TensorDef(T, S.M, S.N, output=True),
106 # def test2(I=TensorDef(T, S.M, S.N),
107 # O=TensorDef(T, S.M, S.N, output=True),
195 # O=TensorDef(U, output=True)):
247 # def test4(O=TensorDef(T, S.M, S.N, output=True),
326 # def test5(value=ScalarDef(T1), O=TensorDef(U, output=True)):
/llvm-project-15.0.7/mlir/python/mlir/dialects/linalg/opdsl/lang/
H A Ddsl.py133 (TensorDef, ScalarDef, IndexAttrDef, UnaryFnAttrDef,
H A Dcomprehension.py476 class TensorDef: class