Lines Matching defs:srcShape

4402 /// Returns true if the `srcShape` or `destShape` is different from the one in
4404 static bool inferStaticShape(PackOp packOp, SmallVectorImpl<int64_t> &srcShape,
4407   srcShape.assign(packOp.getSourceType().getShape().begin(),
4425     if (ShapedType::isDynamic(srcShape[srcPos]) ==
4429     int64_t size = srcShape[srcPos];
4432     srcShape[srcPos] = size;
4461   SmallVector<int64_t> srcShape, destShape;
4462   if (inferStaticShape(packOp, srcShape, destShape)) {
4465     if (srcShape != packOp.getSourceType().getShape()) {
4466       auto newSrcType = packOp.getSourceType().clone(srcShape);
4662 /// Returns true if the `srcShape` or `destShape` is different from the one in
4664 static bool inferStaticShape(UnPackOp op, SmallVectorImpl<int64_t> &srcShape,
4667   srcShape.assign(op.getSourceType().getShape().begin(),
4684     if (ShapedType::isDynamic(srcShape[srcPos]) ==
4688     int64_t size = srcShape[srcPos];
4691     srcShape[srcPos] = size;
4722   SmallVector<int64_t> srcShape, destShape;
4723   if (inferStaticShape(unPackOp, srcShape, destShape)) {
4726     if (srcShape != unPackOp.getSourceType().getShape()) {
4727       auto newSrcType = unPackOp.getSourceType().clone(srcShape);