Lines Matching refs:initOrAllocTensor
175 Value initOrAllocTensor; in splitReduction() local
177 initOrAllocTensor = b.create<bufferization::AllocTensorOp>( in splitReduction()
183 initOrAllocTensor = b.create<linalg::InitTensorOp>( in splitReduction()
188 b.create<linalg::FillOp>(op->getLoc(), constantOp, initOrAllocTensor) in splitReduction()
202 loc, TypeRange({initOrAllocTensor.getType()}), newInputs, in splitReduction()
237 initOrAllocTensor.getDefiningOp(), identityTensor.getDefiningOp<FillOp>(), in splitReduction()
341 Value initOrAllocTensor; in splitReductionByScaling() local
343 initOrAllocTensor = in splitReductionByScaling()
346 initOrAllocTensor = b.create<linalg::InitTensorOp>( in splitReductionByScaling()
351 b.create<linalg::FillOp>(op->getLoc(), constantOp, initOrAllocTensor)); in splitReductionByScaling()
353 initOrAllocTensorOps.push_back(initOrAllocTensor.getDefiningOp()); in splitReductionByScaling()