Searched defs:collapsedShape (Results 1 – 4 of 4) sorted by relevance
| /llvm-project/mlir/lib/Dialect/Utils/ |
| H A D | ReshapeOpsUtils.cpp | 229 ArrayRef<int64_t> collapsedShape, ArrayRef<int64_t> expandedShape, in reshapeLikeShapesAreCompatible() argument
|
| /llvm-project/mlir/lib/Dialect/Linalg/Transforms/ |
| H A D | DropUnitDims.cpp | 867 SmallVector<int64_t> collapsedShape(valType.getShape()); collapseSingletonDimAt() local
|
| H A D | ElementwiseOpFusion.cpp | 582 compute(LinalgOp linalgOp,OpOperand * fusableOpOperand,ArrayRef<AffineMap> reassociationMaps,ArrayRef<int64_t> expandedShape,ArrayRef<int64_t> collapsedShape,PatternRewriter & rewriter) compute() argument
|
| /llvm-project/mlir/lib/Dialect/MemRef/IR/ |
| H A D | MemRefOps.cpp | 2113 verifyCollapsedShape(Operation * op,ArrayRef<int64_t> collapsedShape,ArrayRef<int64_t> expandedShape,ArrayRef<ReassociationIndices> reassociation,bool allowMultipleDynamicDimsPerGroup) verifyCollapsedShape() argument
|