Home
last modified time | relevance | path

Searched refs:dimOrdering (Results 1 – 25 of 28) sorted by relevance

12

/llvm-project-15.0.7/mlir/test/Dialect/SparseTensor/
H A Dsparse_transpose.mlir24 …parse_tensor.encoding<{ dimLevelType = [ "compressed", "compressed" ], dimOrdering = affine_map<(d…
25 …parse_tensor.encoding<{ dimLevelType = [ "compressed", "compressed" ], dimOrdering = affine_map<(d…
26 …parse_tensor.encoding<{ dimLevelType = [ "compressed", "compressed" ], dimOrdering = affine_map<(d…
27 …parse_tensor.encoding<{ dimLevelType = [ "compressed", "compressed" ], dimOrdering = affine_map<(d…
28 …parse_tensor.encoding<{ dimLevelType = [ "compressed", "compressed" ], dimOrdering = affine_map<(d…
29 …parse_tensor.encoding<{ dimLevelType = [ "compressed", "compressed" ], dimOrdering = affine_map<(d…
49 …parse_tensor.encoding<{ dimLevelType = [ "compressed", "compressed" ], dimOrdering = affine_map<(d…
H A Droundtrip_encoding.mlir11 dimOrdering = affine_map<(i,j) -> (i,j)>,
17 …2, #sparse_tensor.encoding<{ dimLevelType = [ "dense", "compressed" ], dimOrdering = affine_map<(d…
H A Dinvalid_encoding.mlir13 #a = #sparse_tensor.encoding<{dimLevelType = ["dense", "compressed"], dimOrdering = affine_map<(i) …
28 #a = #sparse_tensor.encoding<{dimOrdering = "wrong"}> // expected-error {{expected an affine map fo…
33 #a = #sparse_tensor.encoding<{dimOrdering = affine_map<(i,j) -> (i,i)>}> // expected-error {{expect…
H A Dsparse_lower_col.mlir13 dimOrdering = affine_map<(i,j) -> (j,i)>
27 …4, #sparse_tensor.encoding<{ dimLevelType = [ "dense", "compressed" ], dimOrdering = affine_map<(d…
33 …4, #sparse_tensor.encoding<{ dimLevelType = [ "dense", "compressed" ], dimOrdering = affine_map<(d…
34 …4, #sparse_tensor.encoding<{ dimLevelType = [ "dense", "compressed" ], dimOrdering = affine_map<(d…
35 …4, #sparse_tensor.encoding<{ dimLevelType = [ "dense", "compressed" ], dimOrdering = affine_map<(d…
H A Dsparse_expand.mlir8 dimOrdering = affine_map<(i,j) -> (j,i)>
H A Done_shot_bufferize_tensor_copy_insertion.mlir6 dimOrdering = affine_map<(i,j) -> (i,j)>
H A Dsparse_perm.mlir6 dimOrdering = affine_map<(i,j,k) -> (k,i,j)>
H A Dsparse_perm_lower.mlir8 dimOrdering = affine_map<(i,j,k) -> (k,i,j)>
/llvm-project-15.0.7/mlir/test/CAPI/
H A Dsparse_tensor.c37 MlirAffineMap dimOrdering = in testRoundtripEncoding() local
40 mlirAffineMapDump(dimOrdering); in testRoundtripEncoding()
62 ctx, numLevelTypes, levelTypes, dimOrdering, pointerBitWidth, in testRoundtripEncoding()
/llvm-project-15.0.7/mlir/lib/Bindings/Python/
H A DDialectSparseTensor.cpp30 llvm::Optional<MlirAffineMap> dimOrdering, int pointerBitWidth, in populateDialectSparseTensorSubmodule()
34 dimOrdering ? *dimOrdering : MlirAffineMap{nullptr}, in populateDialectSparseTensorSubmodule()
/llvm-project-15.0.7/mlir/lib/CAPI/Dialect/
H A DSparseTensor.cpp40 MlirAffineMap dimOrdering, int pointerBitWidth, int indexBitWidth) { in mlirSparseTensorEncodingAttrGet() argument
47 unwrap(dimOrdering), in mlirSparseTensorEncodingAttrGet()
/llvm-project-15.0.7/mlir/lib/Dialect/SparseTensor/IR/
H A DSparseTensorDialect.cpp145 ArrayRef<DimLevelType> dimLevelType, AffineMap dimOrdering, in verify() argument
151 if (dimOrdering) { in verify()
152 if (!dimOrdering.isPermutation()) in verify()
155 if (dimOrdering.getNumResults() != dimLevelType.size()) in verify()
/llvm-project-15.0.7/mlir/test/Integration/Dialect/SparseTensor/CPU/
H A Dsparse_conversion_sparse2dense.mlir10 dimOrdering = affine_map<(i,j,k) -> (i,j,k)>
15 dimOrdering = affine_map<(i,j,k) -> (j,k,i)>
20 dimOrdering = affine_map<(i,j,k) -> (k,i,j)>
25 dimOrdering = affine_map<(i,j,k) -> (i,j,k)>
30 dimOrdering = affine_map<(i,j,k) -> (j,k,i)>
35 dimOrdering = affine_map<(i,j,k) -> (k,i,j)>
H A Ddense_output.mlir12 dimOrdering = affine_map<(i,j) -> (i,j)>
17 dimOrdering = affine_map<(i,j) -> (i,j)>
H A Dsparse_flatten.mlir22 // Note that any dimOrdering permutation should give the same results
25 dimOrdering = affine_map<(i,j,k,l,m,n,o,p) -> (p,o,j,k,i,l,m,n)>
H A Dsparse_storage.mlir25 dimOrdering = affine_map<(i,j) -> (j,i)>
30 dimOrdering = affine_map<(i,j) -> (j,i)>
39 dimOrdering = affine_map<(i,j) -> (j,i)>
H A Dsparse_conversion_ptr.mlir15 dimOrdering = affine_map<(i,j) -> (j,i)>,
22 dimOrdering = affine_map<(i,j) -> (j,i)>,
H A Dsparse_conversion.mlir9 dimOrdering = affine_map<(i,j,k) -> (i,j,k)>
14 dimOrdering = affine_map<(i,j,k) -> (j,k,i)>
19 dimOrdering = affine_map<(i,j,k) -> (k,i,j)>
H A Dsparse_matmul.mlir8 dimOrdering = affine_map<(i,j) -> (i,j)>
13 dimOrdering = affine_map<(i,j) -> (i,j)>
H A Dsparse_out_simple.mlir21 dimOrdering = affine_map<(i,j) -> (i,j)>
H A Dsparse_conversion_dyn.mlir13 dimOrdering = affine_map<(i,j) -> (j,i)>
H A Dsparse_conversion_sparse2sparse.mlir19 dimOrdering = affine_map<(i,j,k) -> (i,k,j)>
H A Dsparse_transpose.mlir12 dimOrdering = affine_map<(i,j) -> (j,i)>
/llvm-project-15.0.7/mlir/include/mlir-c/Dialect/
H A DSparseTensor.h51 MlirAffineMap dimOrdering, int pointerBitWidth, int indexBitWidth);
/llvm-project-15.0.7/mlir/include/mlir/Dialect/SparseTensor/IR/
H A DSparseTensorAttrDefs.td39 dimOrdering = affine_map<(i,j) -> (j,i)>,
67 "AffineMap":$dimOrdering,

12