Lines Matching refs:resultShape
1706 computeExpandedLayoutMap(MemRefType srcType, ArrayRef<int64_t> resultShape, in computeExpandedLayoutMap() argument
1727 reverseResultStrides.reserve(resultShape.size()); in computeExpandedLayoutMap()
1728 unsigned shapeIndex = resultShape.size() - 1; in computeExpandedLayoutMap()
1736 Wrapper::size(resultShape[shapeIndex--])) in computeExpandedLayoutMap()
1741 resultStrides.resize(resultShape.size(), 1); in computeExpandedLayoutMap()
1747 computeExpandedType(MemRefType srcType, ArrayRef<int64_t> resultShape, in computeExpandedType() argument
1753 return MemRefType::get(resultShape, srcType.getElementType(), layout, in computeExpandedType()
1759 computeExpandedLayoutMap(srcType, resultShape, reassociation); in computeExpandedType()
1763 MemRefType::get(resultShape, srcType.getElementType(), *computedLayout, 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()
1915 return MemRefType::get(resultShape, srcType.getElementType(), layout, in computeCollapsedType()
1927 MemRefType::get(resultShape, srcType.getElementType(), *computedLayout, in computeCollapsedType()
2151 Type SubViewOp::inferRankReducedResultType(ArrayRef<int64_t> resultShape, in inferRankReducedResultType() argument
2159 assert(inferredType.getRank() >= static_cast<int64_t>(resultShape.size()) && in inferRankReducedResultType()
2161 if (inferredType.getRank() == static_cast<int64_t>(resultShape.size())) in inferRankReducedResultType()
2166 computeRankReductionMask(inferredType.getShape(), resultShape); in inferRankReducedResultType()
2175 return MemRefType::get(resultShape, inferredType.getElementType(), map, in inferRankReducedResultType()
2179 Type SubViewOp::inferRankReducedResultType(ArrayRef<int64_t> resultShape, in inferRankReducedResultType() argument
2193 resultShape, sourceRankedTensorType, staticOffsets, staticSizes, in inferRankReducedResultType()