Searched refs:SparseTensorStorage (Results 1 – 3 of 3) sorted by relevance
| /llvm-project/mlir/include/mlir/ExecutionEngine/SparseTensor/ |
| H A D | Storage.h | 195 class SparseTensorStorage final : public SparseTensorStorageBase {
200 SparseTensorStorage(uint64_t dimRank, const uint64_t *dimSizes, in SparseTensorStorage() function
213 SparseTensorStorage(uint64_t dimRank, const uint64_t *dimSizes,
222 SparseTensorStorage(uint64_t dimRank, const uint64_t *dimSizes,
228 static SparseTensorStorage<P, C, V> *
234 static SparseTensorStorage<P, C, V> *
241 static SparseTensorStorage<P, C, V> *
247 ~SparseTensorStorage() final = default;
580 SparseTensorStorage<P, C, V> *SparseTensorStorage<P, C, V>::newEmpty( in newEmpty()
585 return new SparseTensorStorage<P, C, V>(dimRank, dimSizes, lvlRank, lvlSizes, in newEmpty()
[all …]
|
| H A D | File.h | 200 SparseTensorStorage<P, I, V> *
207 auto *tensor = SparseTensorStorage<P, I, V>::newFromCOO(
|
| /llvm-project/mlir/lib/ExecutionEngine/ |
| H A D | SparseTensorRuntime.cpp | 118 return SparseTensorStorage<P, C, V>::newEmpty( \
130 return SparseTensorStorage<P, C, V>::newFromBuffers( \
136 auto &tensor = *static_cast<SparseTensorStorage<P, C, V> *>(ptr); \
|