Home
last modified time | relevance | path

Searched defs:rankDiff (Results 1 – 7 of 7) sorted by relevance

/llvm-project/mlir/lib/Dialect/Vector/Transforms/
H A DVectorInsertExtractStridedSliceRewritePatterns.cpp70 int64_t rankDiff = dstType.getRank() - srcType.getRank(); matchAndRewrite() local
H A DVectorTransforms.cpp350 int64_t rankDiff = in matchAndRewrite() local
1255 int rankDiff = srcType.getRank() - vectorType.getRank(); getTransferFoldableInnerUnitDims() local
/llvm-project/mlir/lib/Conversion/ShapeToStandard/
H A DShapeToStandard.cpp89 Value rankDiff = std::get<1>(tup); in getBroadcastedDim() local
286 Value shape, rankDiff; in matchAndRewrite() local
/llvm-project/mlir/lib/Dialect/Vector/IR/
H A DVectorOps.cpp1673 int64_t rankDiff = broadcastSrcRank - extractResultRank; foldExtractFromBroadcast() local
2245 int64_t rankDiff = dstShape.size() - srcShape.size(); computeBroadcastedUnitDims() local
3051 unsigned rankDiff = destShape.size() - sourceShape.size(); verify() local
3742 unsigned rankDiff = dstRank - srcRank; matchAndRewrite() local
[all...]
/llvm-project/mlir/lib/Dialect/Tosa/IR/
H A DTosaOps.cpp540 auto rankDiff = outShape.size() - shape.getRank(); resolveBroadcastShape() local
/llvm-project/mlir/lib/Dialect/Linalg/Transforms/
H A DVectorization.cpp1107 int32_t rankDiff = dstRank - srcRank; vectorizeTensorExtract() local
/llvm-project/mlir/lib/Dialect/Tensor/IR/
H A DTensorOps.cpp2058 int rankDiff = inferredType.getRank() - desiredResultRank; inferCanonicalRankReducedResultType() local