Home
last modified time | relevance | path

Searched refs:droppedDims (Results 1 – 4 of 4) sorted by relevance

/llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/
H A DDropUnitDims.cpp159 unsigned droppedDims = llvm::count_if( in replaceUnitDimIndexOps() local
161 if (droppedDims != 0) in replaceUnitDimIndexOps()
163 indexOp.dim() - droppedDims); in replaceUnitDimIndexOps()
H A DPromotion.cpp220 llvm::SmallBitVector droppedDims = subView.getDroppedDims(); in promoteSubviewAsNewBuffer() local
223 if (droppedDims[en.index()]) in promoteSubviewAsNewBuffer()
H A DTransforms.cpp213 llvm::SmallBitVector droppedDims = sliceOp.getDroppedDims(); in padOperandToSmallestStaticBoundingBox() local
221 if (droppedDims.test(en.index())) in padOperandToSmallestStaticBoundingBox()
/llvm-project-15.0.7/mlir/lib/Dialect/Tensor/IR/
H A DTensorOps.cpp1135 llvm::SmallBitVector droppedDims(mixedSizes.size()); in getDroppedDims() local
1147 droppedDims.set(size.index()); in getDroppedDims()
1149 return droppedDims; in getDroppedDims()
1157 llvm::SmallBitVector droppedDims = getDroppedDims(); in reifyResultShapes() local
1160 if (droppedDims.test(size.index())) in reifyResultShapes()