Home
last modified time | relevance | path

Searched refs:pidxs (Results 1 – 1 of 1) sorted by relevance

/external/llvm-project/mlir/lib/Dialect/Linalg/Transforms/
DSparsification.cpp253 pidxs(numTensors, std::vector<Value>(numLoops)), in CodeGen()
272 std::vector<std::vector<Value>> pidxs; member
532 args.push_back(codegen.pidxs[tensor][idx]); // position index in genTensorLoad()
640 if (codegen.pidxs[tensor][topSort[pat - 1]]) in genInit()
646 : codegen.pidxs[tensor][topSort[pat - 1]]; in genInit()
647 codegen.pidxs[tensor][idx] = genLoad(rewriter, loc, ptr, p0); in genInit()
701 lo = codegen.pidxs[tensor][idx]; in genFor()
712 codegen.pidxs[tensor][idx] = parOp.getInductionVars()[0]; in genFor()
722 codegen.pidxs[tensor][idx] = forOp.getInductionVar(); in genFor()
743 operands.push_back(codegen.pidxs[tensor][idx]); in genWhile()
[all …]