Home
last modified time | relevance | path

Searched defs:outputDims (Results 1 – 3 of 3) sorted by relevance

/llvm-project/mlir/lib/Dialect/Linalg/Transforms/
H A DConstantFold.cpp159 auto outputDims = getDimPositions(linalgOp.getIndexingMapsArray().back()); in matchAndRewrite() local
/llvm-project/mlir/lib/Dialect/Linalg/IR/
H A DLinalgInterfaces.cpp637 llvm::SmallDenseSet<int64_t> outputDims = findPermutationsIndexingOperand( inferConvolutionDimsImpl() local
796 llvm::SmallDenseSet<int64_t> outputDims = isConvolutionInterfaceImpl() local
1086 llvm::SmallBitVector outputDims(resultShapesFromInputShapesMap.getNumDims()); reifyResultShapes() local
[all...]
/llvm-project/mlir/lib/Dialect/Tensor/IR/
H A DTensorOps.cpp3306 auto outputDims = oldResultType.getShape(); matchAndRewrite() local