Home
last modified time | relevance | path

Searched defs:reductionOp (Results 1 – 13 of 13) sorted by relevance

/llvm-project/mlir/lib/Dialect/Linalg/Transforms/
H A DSplitReduction.cpp68 Operation *reductionOp = combinerOps[0]; in splitReduction() local
399 auto reductionOp = b.create<GenericOp>( in splitReductionByScaling() local
H A DMeshShardingInterfaceImpl.cpp92 std::optional<Operation *> reductionOp = getCombinerOp(op); in getReductionKindOfLinalgOp() local
H A DTilingInterfaceImpl.cpp342 Operation *reductionOp = combinerOps[0]; generateInitialTensorForPartialReduction() local
/llvm-project/mlir/lib/Conversion/AffineToStandard/
H A DAffineToStandard.cpp229 std::optional<arith::AtomicRMWKind> reductionOp = in matchAndRewrite() local
254 std::optional<arith::AtomicRMWKind> reductionOp = in matchAndRewrite() local
/llvm-project/flang/lib/Optimizer/HLFIR/IR/
H A DHLFIROps.cpp492 verifyLogicalReductionOp(LogicalReductionOp reductionOp) verifyLogicalReductionOp() argument
672 verifyArrayAndMaskForReductionOp(NumericalReductionOp reductionOp) verifyArrayAndMaskForReductionOp() argument
711 verifyNumericalReductionOp(NumericalReductionOp reductionOp) verifyNumericalReductionOp() argument
780 verifyCharacterReductionOp(CharacterReductionOp reductionOp) verifyCharacterReductionOp() argument
874 verifyResultForMinMaxLoc(NumericalReductionOp reductionOp) verifyResultForMinMaxLoc() argument
[all...]
/llvm-project/mlir/lib/Conversion/SCFToControlFlow/
H A DSCFToControlFlow.cpp475 auto reductionOp = cast<ReduceOp>(parallelOp.getBody()->getTerminator()); matchAndRewrite() local
/llvm-project/mlir/lib/Dialect/Vector/Transforms/
H A DLowerVectorMultiReduction.cpp387 Operation *reductionOp = rewriter.create<vector::ReductionOp>( matchAndRewrite() local
H A DLowerVectorContract.cpp1156 Operation *reductionOp = in lowerReduction() local
H A DVectorDistribute.cpp1813 auto reductionOp = matchAndRewrite() local
[all...]
/llvm-project/flang/lib/Lower/OpenMP/
H A DReductionProcessor.cpp218 mlir::Value reductionOp; in createScalarCombiner() local
[all...]
/llvm-project/mlir/lib/Conversion/VectorToLLVM/
H A DConvertVectorToLLVM.cpp733 matchAndRewrite(vector::ReductionOp reductionOp,OpAdaptor adaptor,ConversionPatternRewriter & rewriter) const matchAndRewrite() argument
892 auto reductionOp = cast<ReductionOp>(maskableOp.getOperation()); matchAndRewriteMaskableOp() local
[all...]
/llvm-project/mlir/lib/Dialect/Linalg/IR/
H A DLinalgInterfaces.cpp206 Operation *reductionOp = yielded.getDefiningOp(); isContractionBody() local
/llvm-project/mlir/lib/Dialect/Affine/Utils/
H A DUtils.cpp386 Operation *reductionOp = yieldOp->getOperand(i).getDefiningOp(); affineParallelize() local