Lines Matching defs:expandShape
253 /// Compute the expanded sizes of the given \p expandShape for the
263 /// \post result.size() == expandShape.getReassociationIndices()[groupId].size()
269 getExpandedSizes(memref::ExpandShapeOp expandShape, OpBuilder &builder,
272 expandShape.getReassociationIndices()[groupId];
281 MemRefType expandShapeType = expandShape.getResultType();
301 builder, expandShape.getLoc(), s0.floorDiv(productOfAllStaticSizes),
308 /// Compute the expanded strides of the given \p expandShape for the
327 /// \post result.size() == expandShape.getReassociationIndices()[groupId].size()
332 SmallVector<OpFoldResult> getExpandedStrides(memref::ExpandShapeOp expandShape,
338 expandShape.getReassociationIndices()[groupId];
343 MemRefType expandShapeType = expandShape.getResultType();
364 Value source = expandShape.getSrc();
389 builder, expandShape.getLoc(),
402 builder, expandShape.getLoc(), s0 * baseExpandedStride, {origStride});