Searched refs:numTensors (Results 1 – 2 of 2) sorted by relevance
| /external/llvm-project/mlir/lib/Dialect/Linalg/Transforms/ |
| D | Sparsification.cpp | 97 : numTensors(t), numLoops(l), isSparse(t, std::vector<bool>(l, false)) {} in Merger()
109 assert(t < numTensors && i < numLoops); in addLat()
111 latPoints.push_back(LatPoint(numLoops * numTensors, e, numTensors * i + t)); in addLat()
168 if (exp(e).kind == Kind::kTensor && exp(e).e0 == numTensors - 1) in optimize()
207 unsigned tensor(unsigned b) const { return b % numTensors; } in tensor()
208 unsigned index(unsigned b) const { return b / numTensors; } in index()
217 assert(t < numTensors && i < numLoops); in isSparseAccess()
236 const unsigned numTensors; member in __anon4430b2680111::Merger
247 CodeGen(linalg::SparsificationOptions o, unsigned numTensors, in CodeGen()
249 : options(o), loops(numLoops), sizes(numLoops), buffers(numTensors), in CodeGen()
[all …]
|
| /external/llvm-project/mlir/lib/Dialect/Linalg/IR/ |
| D | LinalgOps.cpp | 468 unsigned numTensors = op.getNumInputsAndOutputs(); in verify() local
469 if (sparseAttr.size() != numTensors) in verify()
471 for (unsigned t = 0; t < numTensors; t++) { in verify()
485 if (t == numTensors - 1) in verify()
|