Lines Matching refs:dims
93 SmallVector<unsigned> dims; in splitReduction() local
94 op.getReductionDims(dims); in splitReduction()
95 assert(dims.size() == 1); in splitReduction()
96 unsigned reductionDim = dims[0]; in splitReduction()
284 SmallVector<unsigned> dims; in splitReductionByScaling() local
285 op.getReductionDims(dims); in splitReductionByScaling()
286 if (dims.empty()) in splitReductionByScaling()
289 unsigned reductionDimPos = dims[0]; in splitReductionByScaling()
339 SmallVector<Value> dims = in splitReductionByScaling() local
344 b.create<bufferization::AllocTensorOp>(loc, newT, dims); in splitReductionByScaling()
347 loc, dims, newT.getShape(), t.getElementType()); in splitReductionByScaling()