Searched refs:initOrAllocTensor (Results 1 – 1 of 1) sorted by relevance
175 Value initOrAllocTensor; in splitReduction() local177 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() local343 initOrAllocTensor = in splitReductionByScaling()346 initOrAllocTensor = b.create<linalg::InitTensorOp>( in splitReductionByScaling()351 b.create<linalg::FillOp>(op->getLoc(), constantOp, initOrAllocTensor)); in splitReductionByScaling()[all …]