| /llvm-project/mlir/lib/Dialect/Vector/Transforms/ |
| H A D | LowerVectorMultiReduction.cpp | 166 auto srcShape = multiReductionOp.getSourceVectorType().getShape(); matchAndRewrite() local
329 ArrayRef<int64_t> srcShape = matchAndRewrite() local
439 auto srcShape = srcVectorType.getShape(); matchAndRewrite() local
|
| /llvm-project/mlir/lib/Conversion/TosaToTensor/ |
| H A D | TosaToTensor.cpp | 153 auto srcShape = cast<TensorType>(srcType).getShape(); in createReassociationMapForCollapse() local
[all...] |
| /llvm-project/mlir/lib/Dialect/Tensor/Transforms/ |
| H A D | PackAndUnpackPatterns.cpp | 36 isPackOn1D(RewriterBase & rewriter,Operation * op,ArrayRef<int64_t> srcShape,ArrayRef<int64_t> innerPackTileSize) isPackOn1D() argument
|
| /llvm-project/mlir/lib/Dialect/MemRef/Transforms/ |
| H A D | FoldMemRefAliasOps.cpp | 74 SmallVector<OpFoldResult> srcShape; in resolveSourceIndicesExpandShape() local
|
| H A D | ExpandStridedMetadata.cpp | 509 ArrayRef<int64_t> srcShape = sourceType.getShape(); in getCollapsedStride() local
|
| /llvm-project/mlir/lib/Dialect/Linalg/Transforms/ |
| H A D | DataLayoutPropagation.cpp | 658 ArrayRef<int64_t> srcShape = collapseOp.getSrcType().getShape(); in bubbleUpPackOpThroughCollapseShape() local
|
| H A D | Transforms.cpp | 1212 ArrayRef<int64_t> srcShape = unpackOp.getSourceType().getShape(); matchAndRewrite() local
|
| /llvm-project/mlir/lib/Dialect/SparseTensor/Transforms/Utils/ |
| H A D | CodegenUtils.cpp | 221 ArrayRef<Value> srcShape, ArrayRef<Size> staticDstShape, in genReshapeDstShape() argument
|
| /llvm-project/mlir/lib/Dialect/Tensor/IR/ |
| H A D | TensorOps.cpp | 4160 inferStaticShape(PackOp packOp,SmallVectorImpl<int64_t> & srcShape,SmallVectorImpl<int64_t> & destShape) inferStaticShape() argument
4217 SmallVector<int64_t> srcShape, destShape; canonicalize() local
4394 inferStaticShape(UnPackOp op,SmallVectorImpl<int64_t> & srcShape,SmallVectorImpl<int64_t> & destShape) inferStaticShape() argument
4452 SmallVector<int64_t> srcShape, destShape; canonicalize() local
[all...] |
| /llvm-project/mlir/lib/Dialect/SparseTensor/IR/ |
| H A D | SparseTensorDialect.cpp | 495 SparseTensorEncodingAttr::translateShape(ArrayRef<int64_t> srcShape, in translateShape() argument
|
| /llvm-project/mlir/lib/Dialect/Vector/IR/ |
| H A D | VectorOps.cpp | 2243 computeBroadcastedUnitDims(ArrayRef<int64_t> srcShape,ArrayRef<int64_t> dstShape) computeBroadcastedUnitDims() argument
6273 ArrayRef<int64_t> srcShape = srcType.getShape(); verify() local
|
| /llvm-project/mlir/lib/Dialect/MemRef/IR/ |
| H A D | MemRefOps.cpp | 2406 auto srcShape = srcType.getShape(); computeCollapsedLayoutMap() local
|