| /llvm-project/mlir/lib/ExecutionEngine/SparseTensor/ |
| H A D | Storage.cpp | 20 static inline bool isAllDense(uint64_t lvlRank, const LevelType *lvlTypes) { in isAllDense()
29 const uint64_t *lvlSizes, const LevelType *lvlTypes, in SparseTensorStorageBase()
|
| /llvm-project/mlir/include/mlir/ExecutionEngine/SparseTensor/ |
| H A D | Storage.h | 178 const std::vector<LevelType> lvlTypes; variable
202 const LevelType *lvlTypes, const uint64_t *dim2lvl, in SparseTensorStorage()
582 const uint64_t *lvlSizes, const LevelType *lvlTypes, in newEmpty()
592 const uint64_t *lvlSizes, const LevelType *lvlTypes, in newFromCOO()
603 const uint64_t *lvlSizes, const LevelType *lvlTypes, in newFromBuffers()
619 const uint64_t *lvlSizes, const LevelType *lvlTypes, in SparseTensorStorage()
673 const uint64_t *lvlSizes, const LevelType *lvlTypes, in SparseTensorStorage()
|
| /llvm-project/mlir/test/CAPI/ |
| H A D | sparse_tensor.c | 47 MlirSparseTensorLevelType *lvlTypes = in testRoundtripEncoding() local
|
| /llvm-project/mlir/lib/CAPI/Dialect/ |
| H A D | SparseTensor.cpp | 49 mlirSparseTensorEncodingAttrGet(MlirContext ctx,intptr_t lvlRank,MlirSparseTensorLevelType const * lvlTypes,MlirAffineMap dimToLvl,MlirAffineMap lvlToDim,int posWidth,int crdWidth,MlirAttribute explicitVal,MlirAttribute implicitVal) mlirSparseTensorEncodingAttrGet() argument
|
| /llvm-project/mlir/lib/Bindings/Python/ |
| H A D | DialectSparseTensor.cpp | 46 __anon055f89890102(py::object cls, std::vector<MlirSparseTensorLevelType> lvlTypes, std::optional<MlirAffineMap> dimToLvl, std::optional<MlirAffineMap> lvlToDim, int posWidth, int crdWidth, std::optional<MlirAttribute> explicitVal, std::optional<MlirAttribute> implicitVal, MlirContext context) populateDialectSparseTensorSubmodule() argument
|
| /llvm-project/mlir/lib/ExecutionEngine/ |
| H A D | SparseTensorRuntime.cpp | 173 const LevelType *lvlTypes = MEMREF_GET_PAYLOAD(lvlTypesRef); in _mlir_ciface_newSparseTensor() local
|
| /llvm-project/mlir/lib/Dialect/SparseTensor/IR/ |
| H A D | SparseTensorDialect.cpp | 106 const auto lvlTypes = enc.getLvlTypes(); in foreachField() local
394 ArrayRef<LevelType> lvlTypes = getLvlTypes(); in getBatchLvlRank() local
576 SmallVector<LevelType> lvlTypes; in parse() local
784 verify(function_ref<InFlightDiagnostic ()> emitError,ArrayRef<LevelType> lvlTypes,AffineMap dimToLvl,AffineMap lvlToDim,unsigned posWidth,unsigned crdWidth,Attribute explicitVal,Attribute implicitVal,ArrayRef<SparseTensorDimSliceAttr> dimSlices) verify() argument
1012 SmallVector<LevelType> lvlTypes; getCOOType() local
[all...] |
| /llvm-project/mlir/include/mlir/Dialect/SparseTensor/Utils/ |
| H A D | Merger.h | 639 std::vector<std::vector<LevelType>> lvlTypes; variable
|
| /llvm-project/mlir/lib/Dialect/SparseTensor/Transforms/ |
| H A D | SparseTensorConversion.cpp | 146 SmallVector<Value> lvlTypes; in genLvlTypesBuffer() local
|