Searched refs:newOutputShape (Results 1 – 1 of 1) sorted by relevance
158 SmallVector<int64_t> newOutputShape; in splitReduction() local165 newOutputShape.push_back(ratio); in splitReduction()170 newOutputShape.push_back(oldShape[oldDim]); in splitReduction()179 RankedTensorType::get(newOutputShape, in splitReduction()184 loc, newOutputShape, op.getRegionOutputArgs()[0].getType()); in splitReduction()209 unsigned intermRank = newOutputShape.size(); in splitReduction()