Lines Matching defs:resultShape

2222 computeExpandedLayoutMap(MemRefType srcType, ArrayRef<int64_t> resultShape,
2232   // the proper entries in resultShape.
2243   reverseResultStrides.reserve(resultShape.size());
2244   unsigned shapeIndex = resultShape.size() - 1;
2252            SaturatedInteger::wrap(resultShape[shapeIndex--]))
2257   resultStrides.resize(resultShape.size(), 1);
2262     MemRefType srcType, ArrayRef<int64_t> resultShape,
2268     return MemRefType::get(resultShape, srcType.getElementType(), layout,
2274       computeExpandedLayoutMap(srcType, resultShape, reassociation);
2277   return MemRefType::get(resultShape, srcType.getElementType(), *computedLayout,
2320                           ArrayRef<int64_t> resultShape, Value src,
2325       ExpandShapeOp::computeExpandedType(srcType, resultShape, reassociation);
2333                           ArrayRef<int64_t> resultShape, Value src,
2339       ExpandShapeOp::computeExpandedType(srcType, resultShape, reassociation);
2492   SmallVector<int64_t> resultShape;
2493   resultShape.reserve(reassociation.size());
2499     resultShape.push_back(groupSize.asInteger());
2506     return MemRefType::get(resultShape, srcType.getElementType(), layout,
2517   return MemRefType::get(resultShape, srcType.getElementType(), *computedLayout,
2759 Type SubViewOp::inferRankReducedResultType(ArrayRef<int64_t> resultShape,
2766   assert(inferredType.getRank() >= static_cast<int64_t>(resultShape.size()) &&
2768   if (inferredType.getRank() == static_cast<int64_t>(resultShape.size()))
2773       computeRankReductionMask(inferredType.getShape(), resultShape);
2779   rankReducedStrides.reserve(resultShape.size());
2784   return MemRefType::get(resultShape, inferredType.getElementType(),
2791 Type SubViewOp::inferRankReducedResultType(ArrayRef<int64_t> resultShape,
2802       resultShape, sourceRankedTensorType, staticOffsets, staticSizes,