Searched refs:getNumInitTensors (Results 1 – 7 of 7) sorted by relevance
46 static unsigned getNumInitTensors() { return 0; } in getNumInitTensors() function101 unsigned getNumInitTensors() { in getNumInitTensors() function
147 /*methodName=*/"getNumInitTensors"416 return {base, base + $_op.getNumInitTensors()};428 assert(i < $_op.getNumInitTensors() && "overflowing init tensor index");484 return getNumInputsAndOutputBuffers() + $_op.getNumInitTensors();
394 assert(op.getNumInitTensors() == iterArgs.size() && in tileLinalgOpImpl()417 assert(op.getNumInitTensors() == op->getNumResults() && in tileLinalgOpImpl()425 for (auto idx : llvm::seq<unsigned>(0, op.getNumInitTensors())) in tileLinalgOpImpl()432 for (unsigned idx = 0, e = op.getNumInitTensors(); idx != e; ++idx) { in tileLinalgOpImpl()
68 bool hasInitTensor = resultIndex < linalgOp.getNumInitTensors(); in allocateBuffersForResults()134 for (Value v : ValueRange(outputs).drop_front(genericOp.getNumInitTensors())) in finalizeBufferAllocationForGenericOp()
136 if (linalgOp.getNumInitTensors() != linalgOp->getNumResults()) in matchAndRewriteBase()
486 if (t == numInputs && !op.getNumInitTensors()) { in genBuffers()
107 if (linalgOp.getNumInitTensors()) in getStaticShape()