Searched refs:mixedStrides (Results 1 – 6 of 6) sorted by relevance
| /llvm-project-15.0.7/mlir/include/mlir/Dialect/Arithmetic/Utils/ |
| H A D | Utils.h | 56 SmallVector<OpFoldResult> mixedStrides(op.getMixedStrides()); in matchAndRewrite()
59 canonicalizeSubViewPart(mixedStrides, ShapedType::isDynamicStrideOrOffset); in matchAndRewrite()
64 resultTypeFunc(op, mixedOffsets, mixedSizes, mixedStrides); in matchAndRewrite()
69 mixedOffsets, mixedSizes, mixedStrides); in matchAndRewrite()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/MemRef/Transforms/ |
| H A D | FoldSubViewOps.cpp | 49 SmallVector<OpFoldResult> mixedStrides = subViewOp.getMixedStrides(); in resolveSourceIndices() local
72 if (auto attr = mixedStrides[index].dyn_cast<Attribute>()) { in resolveSourceIndices()
75 dynamicOperands.push_back(mixedStrides[index].get<Value>()); in resolveSourceIndices()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Tensor/Transforms/ |
| H A D | BufferizableOpInterfaceImpl.cpp | 283 SmallVector<OpFoldResult> mixedStrides = extractSliceOp.getMixedStrides(); in bufferize() local
297 mixedSizes, mixedStrides) in bufferize()
301 mixedStrides); in bufferize()
708 SmallVector<OpFoldResult> mixedStrides = insertSliceOp.getMixedStrides(); in bufferize() local
722 mixedOffsets, mixedSizes, mixedStrides) in bufferize()
726 mixedStrides); in bufferize()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Tensor/IR/ |
| H A D | TensorOps.cpp | 1355 ArrayRef<OpFoldResult> mixedStrides) { in operator ()()
1358 mixedStrides); in operator ()()
1609 SmallVector<OpFoldResult> mixedStrides(insertSliceOp.getMixedStrides()); in matchAndRewrite() local
1612 canonicalizeSubViewPart(mixedStrides, ShapedType::isDynamicStrideOrOffset); in matchAndRewrite()
1617 mixedOffsets, mixedSizes, mixedStrides); in matchAndRewrite()
1631 mixedSizes, mixedStrides); in matchAndRewrite()
2315 SmallVector<OpFoldResult> mixedStrides(insertSliceOp.getMixedStrides()); in matchAndRewrite() local
2318 canonicalizeSubViewPart(mixedStrides, ShapedType::isDynamicStrideOrOffset); in matchAndRewrite()
2325 mixedStrides); in matchAndRewrite()
2335 mixedSizes, mixedStrides); in matchAndRewrite()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/MemRef/IR/ |
| H A D | MemRefOps.cpp | 2440 ArrayRef<OpFoldResult> mixedSizes, ArrayRef<OpFoldResult> mixedStrides) { in getCanonicalSubViewResultType() argument
2442 mixedSizes, mixedStrides) in getCanonicalSubViewResultType()
2469 ArrayRef<OpFoldResult> mixedStrides) { in getCanonicalSubViewResultType() argument
2472 mixedStrides); in getCanonicalSubViewResultType()
2485 auto mixedStrides = subViewOp.getMixedStrides(); in isTrivialSubViewOp() local
2495 if (llvm::any_of(mixedStrides, [](OpFoldResult ofr) { in isTrivialSubViewOp()
2597 ArrayRef<OpFoldResult> mixedStrides) { in operator ()()
2600 mixedStrides); in operator ()()
|
| /llvm-project-15.0.7/mlir/lib/Conversion/MemRefToLLVM/ |
| H A D | MemRefToLLVM.cpp | 1734 SmallVector<OpFoldResult> mixedStrides = subViewOp.getMixedStrides(); in matchAndRewrite() local
1735 assert(mixedSizes.size() == mixedStrides.size() && in matchAndRewrite()
|