Searched refs:setDim (Results 1 – 3 of 3) sorted by relevance
| /llvm-project-15.0.7/mlir/unittests/Dialect/SparseTensor/ |
| H A D | MergerTest.cpp | 311 merger.setDim(t0, l0, Dim::kSparse); in MergerTest3T1L()
315 merger.setDim(t1, l0, Dim::kSparse); in MergerTest3T1L()
319 merger.setDim(t2, l0, Dim::kDense); in MergerTest3T1L()
334 merger.setDim(t0, l0, Dim::kSparse); in MergerTest4T1L()
338 merger.setDim(t1, l0, Dim::kSparse); in MergerTest4T1L()
342 merger.setDim(t2, l0, Dim::kSparse); in MergerTest4T1L()
346 merger.setDim(t3, l0, Dim::kDense); in MergerTest4T1L()
365 merger.setDim(t0, l0, Dim::kSparse); in MergerTest3T1LD()
369 merger.setDim(t1, l0, Dim::kDense); in MergerTest3T1LD()
373 merger.setDim(t2, l0, Dim::kDense); in MergerTest3T1LD()
|
| /llvm-project-15.0.7/mlir/include/mlir/Dialect/SparseTensor/Utils/ |
| H A D | Merger.h | 250 void setDim(unsigned t, unsigned i, Dim d) { dims[t][i] = d; } in setDim() function
|
| /llvm-project-15.0.7/mlir/lib/Dialect/SparseTensor/Transforms/ |
| H A D | Sparsification.cpp | 167 merger.setDim(tensor, idx, dim); in findAffine()
|