Home
last modified time | relevance | path

Searched defs:mixedSizes (Results 1 – 9 of 9) sorted by relevance

/llvm-project/mlir/lib/Dialect/Tensor/Transforms/
H A DReshapePatterns.cpp42 SmallVector<OpFoldResult> mixedSizes = extractSliceOp.getMixedSizes(); in matchAndRewrite() local
107 SmallVector<OpFoldResult> mixedSizes = insertSliceOp.getMixedSizes(); in matchAndRewrite() local
H A DBufferizableOpInterfaceImpl.cpp374 SmallVector<OpFoldResult> mixedSizes = extractSliceOp.getMixedSizes(); bufferize() local
407 SmallVector<OpFoldResult> mixedSizes = extractSliceOp.getMixedSizes(); getBufferType() local
685 SmallVector<OpFoldResult> mixedSizes = insertSliceOp.getMixedSizes(); bufferize() local
/llvm-project/mlir/include/mlir/Interfaces/
H A DValueBoundsOpInterface.h47 SmallVector<OpFoldResult> mixedSizes; variable
/llvm-project/mlir/lib/Dialect/NVGPU/TransformOps/
H A DNVGPUTransformOps.cpp997 prodExprInBytes, mixedSizes); in buildBarrierArriveTx() argument
954 SmallVector<OpFoldResult> mixedSizes = buildGlobalMemRefDescriptor() local
984 auto mixedSizes = memref::getMixedSizes(rewriter, loc, sharedMemref); buildTmaAsyncLoad() local
/llvm-project/mlir/lib/Dialect/Tensor/IR/
H A DTensorOps.cpp90 SmallVector<OpFoldResult> mixedSizes; getOrCreateDestination() local
136 getDroppedDims(ArrayRef<int64_t> reducedShape,ArrayRef<OpFoldResult> mixedSizes) getDroppedDims() argument
2209 SmallVector<OpFoldResult> mixedSizes = getMixedSizes(); reifyResultShapes() local
2642 SmallVector<OpFoldResult> mixedSizes(insertSliceOp.getMixedSizes()); matchAndRewrite() local
2738 SmallVector<OpFoldResult> mixedSizes(insertSliceOp.getMixedSizes()); matchAndRewrite() local
4052 SmallVector<OpFoldResult> mixedSizes; createDestinationTensor() local
4360 SmallVector<OpFoldResult> mixedSizes; createDestinationTensor() local
[all...]
/llvm-project/mlir/lib/Dialect/Vector/Transforms/
H A DVectorTransferOpTransforms.cpp261 getReducedShape(ArrayRef<OpFoldResult> mixedSizes) getReducedShape() argument
/llvm-project/mlir/lib/Dialect/Linalg/Utils/
H A DUtils.cpp886 getReassociationMapForFoldingUnitDims(ArrayRef<OpFoldResult> mixedSizes) getReassociationMapForFoldingUnitDims() argument
/llvm-project/mlir/lib/Dialect/MemRef/IR/
H A DMemRefOps.cpp3049 getCanonicalSubViewResultType(MemRefType currentResultType,MemRefType currentSourceType,MemRefType sourceType,ArrayRef<OpFoldResult> mixedOffsets,ArrayRef<OpFoldResult> mixedSizes,ArrayRef<OpFoldResult> mixedStrides) getCanonicalSubViewResultType() argument
3116 auto mixedSizes = subViewOp.getMixedSizes(); isTrivialSubViewOp() local
/llvm-project/mlir/lib/Dialect/Linalg/Transforms/
H A DElementwiseOpFusion.cpp2030 SmallVector<OpFoldResult> mixedSizes = matchAndRewrite() local