Lines Matching refs:pidxs
63 pidxs(numTensors, std::vector<Value>(numLoops)), in CodeGen()
84 std::vector<std::vector<Value>> pidxs; member
707 assert(codegen.pidxs[tensor][idx] != nullptr); in genSubscript()
708 args.push_back(codegen.pidxs[tensor][idx]); // position index in genSubscript()
1099 if (codegen.pidxs[tensor][topSort[pat - 1]]) in genInit()
1105 : codegen.pidxs[tensor][topSort[pat - 1]]; in genInit()
1106 codegen.pidxs[tensor][idx] = genLoad(codegen, builder, loc, ptr, p0); in genInit()
1211 Value lo = isSparse ? codegen.pidxs[tensor][idx] : codegen.loops[idx]; in genFor()
1225 codegen.pidxs[tensor][idx] = parOp.getInductionVars()[0]; in genFor()
1253 codegen.pidxs[tensor][idx] = iv; in genFor()
1276 operands.push_back(codegen.pidxs[tensor][idx]); in genWhile()
1313 codegen.pidxs[tensor][idx] = after->getArgument(o++); in genWhile()
1357 Value s = codegen.pidxs[tensor][idx]; in genLocals()
1388 if (codegen.pidxs[tensor][topSort[pat - 1]]) in genLocals()
1391 : codegen.pidxs[tensor][topSort[pat - 1]]; in genLocals()
1392 codegen.pidxs[tensor][idx] = genAddress( in genLocals()
1446 Value op3 = codegen.pidxs[tensor][idx]; in genWhileInduction()
1451 codegen.pidxs[tensor][idx] = whileOp->getResult(o++); in genWhileInduction()