Home
last modified time | relevance | path

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

/llvm-project-15.0.7/mlir/lib/Dialect/Utils/
H A DReshapeOpsUtils.cpp228 ArrayRef<int64_t> collapsedShape, ArrayRef<int64_t> expandedShape, in reshapeLikeShapesAreCompatible() argument
250 if (!ShapedType::isDynamic(collapsedShape[map.index()])) { in reshapeLikeShapesAreCompatible()
257 if (collapsedShape[map.index()] != linearizedStaticShape) { in reshapeLikeShapesAreCompatible()
/llvm-project-15.0.7/mlir/lib/Dialect/MemRef/IR/
H A DMemRefOps.cpp1613 verifyCollapsedShape(Operation *op, ArrayRef<int64_t> collapsedShape, in verifyCollapsedShape() argument
1618 if (collapsedShape.size() != reassociation.size()) in verifyCollapsedShape()
1620 << reassociation.size() << ", expected " << collapsedShape.size(); in verifyCollapsedShape()
1648 if (ShapedType::isDynamic(collapsedShape[collapsedDim]) != foundDynamic) in verifyCollapsedShape()
1660 if (groupSize != collapsedShape[collapsedDim]) in verifyCollapsedShape()
1662 << collapsedShape[collapsedDim] in verifyCollapsedShape()
1667 if (collapsedShape.empty()) { in verifyCollapsedShape()
/llvm-project-15.0.7/mlir/include/mlir/Dialect/Utils/
H A DReshapeOpsUtils.h152 ArrayRef<int64_t> collapsedShape, ArrayRef<int64_t> expandedShape,
/llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/
H A DElementwiseOpFusion.cpp486 ArrayRef<int64_t> collapsedShape,
515 ArrayRef<int64_t> collapsedShape, in compute() argument