Home
last modified time | relevance | path

Searched refs:computeRankReductionMask (Results 1 – 5 of 5) sorted by relevance

/llvm-project-15.0.7/mlir/include/mlir/IR/
H A DBuiltinTypes.h343 computeRankReductionMask(ArrayRef<int64_t> originalShape,
/llvm-project-15.0.7/mlir/lib/IR/
H A DBuiltinTypes.cpp438 mlir::computeRankReductionMask(ArrayRef<int64_t> originalShape, in computeRankReductionMask() function in mlir
483 computeRankReductionMask(originalShape, candidateReducedShape); in isRankReducedType()
/llvm-project-15.0.7/mlir/include/mlir/Dialect/Tensor/IR/
H A DTensorOps.td300 computeRankReductionMask() {
301 return ::mlir::computeRankReductionMask(getSourceType().getShape(),
/llvm-project-15.0.7/mlir/lib/Dialect/Tensor/IR/
H A DTensorOps.cpp260 auto dimMask = computeRankReductionMask( in matchAndRewrite()
/llvm-project-15.0.7/mlir/lib/Dialect/MemRef/IR/
H A DMemRefOps.cpp2166 computeRankReductionMask(inferredType.getShape(), resultShape); in inferRankReducedResultType()