Home
last modified time | relevance | path

Searched refs:resultShape (Results 1 – 19 of 19) sorted by relevance

/llvm-project-15.0.7/mlir/lib/Dialect/
H A DTraits.cpp70 resultShape.clear(); in getBroadcastedShape()
79 auto iR = resultShape.rbegin(); in getBroadcastedShape()
106 resultShape.clear(); in getBroadcastedShape()
175 SmallVector<int64_t, 4> resultShape; in getBroadcastedType() local
181 return VectorType::get(resultShape, elementType); in getBroadcastedType()
183 return RankedTensorType::get(resultShape, elementType); in getBroadcastedType()
252 SmallVector<int64_t, 4> resultShape; in verifyCompatibleOperandBroadcast() local
254 resultShape); in verifyCompatibleOperandBroadcast()
256 SmallVector<int64_t, 4> temp = resultShape; in verifyCompatibleOperandBroadcast()
270 getShape(type).take_back(resultShape.size()); in verifyCompatibleOperandBroadcast()
[all …]
/llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/
H A DFusePadOpWithLinalgProducer.cpp64 ReifiedRankedShapedTypeDims resultShape; in matchAndRewrite() local
68 resultShape)) || in matchAndRewrite()
69 resultShape.size() != 1) { in matchAndRewrite()
78 auto resultSizes = getAsOpFoldResult(resultShape[0]); in matchAndRewrite()
/llvm-project-15.0.7/flang/lib/Evaluate/
H A Dfold-reduction.h93 ConstantSubscripts resultShape; // empty -> scalar in DoReduction() local
95 resultShape = array.shape(); in DoReduction()
96 resultShape.erase(resultShape.begin() + (*dim - 1)); in DoReduction()
100 for (auto n{GetSize(resultShape)}; n-- > 0; in DoReduction()
117 std::move(elements), std::move(resultShape)}; in DoReduction()
119 return {std::move(elements), std::move(resultShape)}; in DoReduction()
H A Dfold-integer.cpp314 ConstantSubscripts at{array->lbounds()}, maskAt, resultIndices, resultShape; in Test() local
334 resultShape = array->shape(); in Test()
335 resultShape.erase( in Test()
336 resultShape.begin() + zbDim); // scalar if array is vector in Test()
338 ConstantSubscript n{GetSize(resultShape)}; in Test()
369 resultShape = ConstantSubscripts{array->Rank()}; // always a vector in Test()
390 std::move(resultElements), std::move(resultShape)}; in Test()
H A Dshape.cpp687 auto resultShape{(*this)(subp.result())}; in operator ()() local
688 if (resultShape && !useResultSymbolShape_) { in operator ()()
693 for (auto &extent : *resultShape) { in operator ()()
699 return resultShape; in operator ()()
H A Dfold-implementation.h200 ConstantSubscripts resultShape; in ApplySubscripts() local
205 resultShape.push_back(static_cast<ConstantSubscript>(ss.size())); in ApplySubscripts()
219 CHECK(k < GetRank(resultShape)); in ApplySubscripts()
223 if (++ssAt[k] == resultShape[k]) { in ApplySubscripts()
240 CHECK(k == GetRank(resultShape)); in ApplySubscripts()
243 return Constant<T>{array.LEN(), std::move(values), std::move(resultShape)}; in ApplySubscripts()
246 std::move(resultShape)}; in ApplySubscripts()
248 return Constant<T>{std::move(values), std::move(resultShape)}; in ApplySubscripts()
H A Dcheck-expression.cpp440 } else if (auto resultShape{GetShape(context, folded)}) { in NonPointerInitializationExpr() local
442 *resultShape, CheckConformanceFlags::None, in NonPointerInitializationExpr()
/llvm-project-15.0.7/mlir/lib/Dialect/MemRef/Transforms/
H A DResolveShapedTypeResultDims.cpp53 Value resultShape = reifiedResultShapes[dimValue.getResultNumber()]; in matchAndRewrite() local
54 auto resultShapeType = resultShape.getType().dyn_cast<RankedTensorType>(); in matchAndRewrite()
60 dimOp, resultShape, in matchAndRewrite()
/llvm-project-15.0.7/mlir/lib/Dialect/MemRef/IR/
H A DMemRefOps.cpp1727 reverseResultStrides.reserve(resultShape.size()); in computeExpandedLayoutMap()
1728 unsigned shapeIndex = resultShape.size() - 1; in computeExpandedLayoutMap()
1741 resultStrides.resize(resultShape.size(), 1); in computeExpandedLayoutMap()
1747 computeExpandedType(MemRefType srcType, ArrayRef<int64_t> resultShape, in computeExpandedType() argument
1759 computeExpandedLayoutMap(srcType, resultShape, reassociation); in computeExpandedType()
1769 ArrayRef<int64_t> resultShape, Value src, in build() argument
1774 computeExpandedType(srcType, resultShape, reassociation); in build()
1901 SmallVector<int64_t> resultShape; in computeCollapsedType() local
1902 resultShape.reserve(reassociation.size()); in computeCollapsedType()
1908 resultShape.push_back(groupSize.asSize()); in computeCollapsedType()
[all …]
/llvm-project-15.0.7/mlir/include/mlir/Dialect/Utils/
H A DReshapeOpsUtils.h338 ArrayRef<int64_t> srcShape, ArrayRef<int64_t> resultShape) const { in findCollapsingReassociation()
342 return {getReassociationIndicesForCollapse(srcShape, resultShape)}; in findCollapsingReassociation()
349 resultShape.slice(resultIndices.front(), resultIndices.size()); in findCollapsingReassociation()
/llvm-project-15.0.7/mlir/include/mlir/Dialect/
H A DTraits.h48 SmallVectorImpl<int64_t> &resultShape);
/llvm-project-15.0.7/mlir/lib/Dialect/Tensor/IR/
H A DTensorInferTypeOpInterfaceImpl.cpp156 auto resultShape = getReshapeOutputShapeFromInputShape( in reifyResultShapes() local
159 reifiedReturnShapes.push_back(getAsValues(b, loc, resultShape)); in reifyResultShapes()
H A DTensorOps.cpp1133 ArrayRef<int64_t> resultShape = getType().getShape(); in getDroppedDims() local
1143 (shapePos < resultShape.size() && resultShape[shapePos] == 1)) { in getDroppedDims()
1868 ArrayRef<int64_t> resultShape) { in inferResultType() argument
1872 assert((resultShape.empty() || resultShape.size() == rank) && in inferResultType()
1880 inferredShape.push_back(resultShape.empty() ? ShapedType::kDynamicSize in inferResultType()
1881 : resultShape[i]); in inferResultType()
1884 assert((resultShape.empty() || size == resultShape[i] || in inferResultType()
1885 resultShape[i] == ShapedType::kDynamicSize) && in inferResultType()
/llvm-project-15.0.7/mlir/lib/Dialect/Shape/IR/
H A DShape.cpp659 SmallVector<int64_t, 6> resultShape; in fold() local
663 if (!OpTrait::util::getBroadcastedShape(lhsShape, rhsShape, resultShape)) in fold()
667 return builder.getIndexTensorAttr(resultShape); in fold()
858 SmallVector<int64_t, 6> resultShape; in fold() local
859 resultShape.append(lhsShape.begin(), lhsShape.end()); in fold()
860 resultShape.append(rhsShape.begin(), rhsShape.end()); in fold()
862 return builder.getIndexTensorAttr(resultShape); in fold()
/llvm-project-15.0.7/mlir/lib/Conversion/TosaToLinalg/
H A DTosaToLinalgNamed.cpp374 auto resultShape = resultTy.getShape(); in matchAndRewrite() local
416 RankedTensorType::get({resultShape[0], resultShape[1], resultShape[2], in matchAndRewrite()
/llvm-project-15.0.7/mlir/lib/Dialect/Vector/IR/
H A DVectorOps.cpp3849 SmallVector<int64_t> resultShape = applyPermutationMap( in matchAndRewrite() local
3851 if (transferOp.getMask() || !vectorShape.equals(resultShape)) { in matchAndRewrite()
4289 auto resultShape = resultVectorType.getShape(); in verifyVectorShapeCast() local
4295 resultShape.begin(), resultShape.end(), 1LL, std::multiplies<int64_t>{}); in verifyVectorShapeCast()
4303 if (!isValidShapeCast(sourceShape, resultShape)) in verifyVectorShapeCast()
4306 if (!isValidShapeCast(resultShape, sourceShape)) in verifyVectorShapeCast()
4730 auto resultShape = resultType.getShape(); in verify() local
4734 if (attrValue < 0 || attrValue > resultShape[it.index()]) in verify()
/llvm-project-15.0.7/mlir/lib/Dialect/Linalg/IR/
H A DLinalgOps.cpp1397 ArrayRef<int64_t> resultShape = resultType.getShape(); in matchAndRewrite() local
1401 assert(resultShape.size() == currMixedSizes.size() && in matchAndRewrite()
1403 for (auto it : llvm::zip(resultShape, currMixedSizes)) { in matchAndRewrite()
/llvm-project-15.0.7/mlir/include/mlir/Dialect/MemRef/IR/
H A DMemRefOps.td1273 OpBuilder<(ins "ArrayRef<int64_t>":$resultShape, "Value":$src,
1656 static Type inferRankReducedResultType(ArrayRef<int64_t> resultShape,
1661 static Type inferRankReducedResultType(ArrayRef<int64_t> resultShape,
/llvm-project-15.0.7/mlir/include/mlir/Dialect/Tensor/IR/
H A DTensorOps.td994 ArrayRef<int64_t> resultShape = {});