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()