Home
last modified time | relevance | path

Searched defs:resultShape (Results 1 – 24 of 24) sorted by relevance

/llvm-project/mlir/lib/Dialect/
H A DTraits.cpp62 SmallVectorImpl<int64_t> &resultShape) { in getBroadcastedShape() argument
176 SmallVector<int64_t, 4> resultShape; in getBroadcastedType() local
250 SmallVector<int64_t, 4> resultShape; verifyCompatibleOperandBroadcast() local
[all...]
/llvm-project/mlir/lib/Dialect/Linalg/Transforms/
H A DFusePadOpWithLinalgProducer.cpp64 ReifiedRankedShapedTypeDims resultShape; in matchAndRewrite() local
H A DVectorization.cpp1798 ArrayRef<int64_t> resultShape = unpackOp.getDestType().getShape(); vectorizeUnPackOpPrecondition() local
/llvm-project/mlir/lib/Dialect/MemRef/Transforms/
H A DResolveShapedTypeResultDims.cpp63 Value resultShape = reifiedResultShapes[dimValue.getResultNumber()]; matchAndRewrite() local
/llvm-project/flang/lib/Optimizer/HLFIR/Transforms/
H A DSimplifyHLFIRIntrinsics.cpp46 mlir::Value resultShape = genResultShape(loc, builder, array); matchAndRewrite() local
[all...]
/llvm-project/flang/lib/Optimizer/HLFIR/IR/
H A DHLFIROps.cpp524 llvm::ArrayRef<int64_t> resultShape = resultExpr.getShape(); verifyLogicalReductionOp() local
590 llvm::ArrayRef<int64_t> resultShape = resultExpr.getShape(); verify() local
744 llvm::ArrayRef<int64_t> resultShape = resultExpr.getShape(); verifyNumericalReductionOp() local
809 llvm::ArrayRef<int64_t> resultShape = resultExpr.getShape(); verifyCharacterReductionOp() local
897 llvm::ArrayRef<int64_t> resultShape = resultExpr.getShape(); verifyResultForMinMaxLoc() local
1059 llvm::ArrayRef<int64_t> resultShape = resultTy.getShape(); verify() local
1181 llvm::ArrayRef<int64_t> resultShape = resultTy.getShape(); verify() local
1229 llvm::ArrayRef<int64_t> resultShape = resultTy.getShape(); verify() local
[all...]
/llvm-project/flang/lib/Evaluate/
H A Dfold-reduction.h188 ConstantSubscripts resultShape; // empty -> scalar DoReduction() local
H A Dfold-implementation.h207 ConstantSubscripts resultShape; ApplySubscripts() local
H A Dcheck-expression.cpp444 } else if (auto resultShape{GetShape(context, folded)}) { NonPointerInitializationExpr() local
H A Dshape.cpp771 auto resultShape{(*this)(subp.result())}; operator ()() local
H A Dfold-integer.cpp365 ConstantSubscripts at{array->lbounds()}, maskAt, resultIndices, resultShape; Test() local
/llvm-project/flang/lib/Lower/
H A DHlfirIntrinsics.cpp341 hlfir::ExprType::Shape resultShape; lowerImpl() local
H A DConvertExprToHLFIR.cpp188 mlir::Value resultShape; member
[all...]
/llvm-project/mlir/lib/Conversion/TosaToTensor/
H A DTosaToTensor.cpp60 auto resultShape = llvm::map_to_vector(newShape, [&](int64_t size) -> int64_t { in inferReshapeExpandedType() local
/llvm-project/mlir/lib/Dialect/MemRef/IR/
H A DMemRefOps.cpp2206 computeExpandedLayoutMap(MemRefType srcType,ArrayRef<int64_t> resultShape,ArrayRef<ReassociationIndices> reassociation) computeExpandedLayoutMap() argument
2246 computeExpandedType(MemRefType srcType,ArrayRef<int64_t> resultShape,ArrayRef<ReassociationIndices> reassociation) computeExpandedType() argument
2304 build(OpBuilder & builder,OperationState & result,ArrayRef<int64_t> resultShape,Value src,ArrayRef<ReassociationIndices> reassociation) build() argument
2317 build(OpBuilder & builder,OperationState & result,ArrayRef<int64_t> resultShape,Value src,ArrayRef<ReassociationIndices> reassociation,ArrayRef<OpFoldResult> outputShape) build() argument
2471 SmallVector<int64_t> resultShape; computeCollapsedType() local
2738 inferRankReducedResultType(ArrayRef<int64_t> resultShape,MemRefType sourceRankedTensorType,ArrayRef<int64_t> offsets,ArrayRef<int64_t> sizes,ArrayRef<int64_t> strides) inferRankReducedResultType() argument
2770 inferRankReducedResultType(ArrayRef<int64_t> resultShape,MemRefType sourceRankedTensorType,ArrayRef<OpFoldResult> offsets,ArrayRef<OpFoldResult> sizes,ArrayRef<OpFoldResult> strides) inferRankReducedResultType() argument
[all...]
/llvm-project/mlir/lib/Dialect/Tensor/IR/
H A DTensorOps.cpp1021 ArrayRef<int64_t> resultShape = resultType.getShape(); matchAndRewrite() local
1278 SmallVector<int64_t> resultShape(indicesType.getShape().drop_back()); inferResultType() local
2943 inferResultType(RankedTensorType sourceType,ArrayRef<int64_t> staticLow,ArrayRef<int64_t> staticHigh,ArrayRef<int64_t> resultShape) inferResultType() argument
3972 SmallVector<int64_t> resultShape = llvm::to_vector(sourceShape); getPackOpResultTypeShape() local
4036 SmallVector<int64_t> resultShape = getPackOpResultTypeShape( inferPackedType() local
[all...]
H A DTensorTilingInterfaceImpl.cpp82 ReifiedRankedShapedTypeDims resultShape; getPackUnPackIterationDomain() local
/llvm-project/mlir/lib/Dialect/Shape/IR/
H A DShape.cpp664 SmallVector<int64_t, 6> resultShape; in fold() local
865 SmallVector<int64_t, 6> resultShape; fold() local
/llvm-project/mlir/lib/Dialect/Arith/Transforms/
H A DEmulateWideInt.cpp156 llvm::ArrayRef<int64_t> resultShape = resultType.getShape(); in constructResultVector() local
/llvm-project/mlir/lib/Conversion/TosaToLinalg/
H A DTosaToLinalgNamed.cpp448 auto resultShape = resultTy.getShape(); matchAndRewrite() local
H A DTosaToLinalg.cpp612 SmallVector<int64_t> resultShape; expandRank() local
/llvm-project/mlir/lib/Dialect/Mesh/IR/
H A DMeshOps.cpp693 SmallVector<int64_t> resultShape = sliceResultType() local
/llvm-project/flang/lib/Optimizer/Transforms/
H A DSimplifyIntrinsics.cpp875 fir::SequenceType::Shape resultShape(1, rank); genRuntimeMinMaxlocBody() local
/llvm-project/mlir/lib/Dialect/Vector/IR/
H A DVectorOps.cpp4747 SmallVector<int64_t> resultShape = applyPermutationMap( matchAndRewrite() local
5207 auto resultShape = resultVectorType.getShape(); verifyVectorShapeCast() local
5823 auto resultShape = resultType.getShape(); verify() local
[all...]