Lines Matching refs:getShape
132 vectorType.getShape() == ArrayRef<int64_t>{1}) in getTransferMinorIdentityMap()
355 return llvm::to_vector<4>(getSourceVectorType().getShape()); in getShapeForUnroll()
361 for (auto it : llvm::enumerate(getSourceVectorType().getShape())) in verify()
487 return llvm::to_vector<4>(getVectorType().getShape()); in getShapeForUnroll()
693 extents[pos] = getAffineConstantExpr(v.getShape()[idx], ctx); in verifyOutputShape()
745 unsigned rank = vectorType ? vectorType.getShape().size() : 0; in verify()
786 if (lhsMaskType.getShape().size() != lhsType.getShape().size() || in verify()
787 rhsMaskType.getShape().size() != rhsType.getShape().size()) in verify()
834 auto lhsShape = getLhsType().getShape(); in getIterationBounds()
853 iterationBounds.push_back(resVectorType.getShape()[resDimIndex]); in getIterationBounds()
1033 vectorType.getShape().drop_front(n), vectorType.getElementType())); in inferReturnTypes()
1042 return vectorType && vectorType.getShape().equals({1}) && in isCompatibleReturnTypes()
1333 extractVecType.getShape() != in foldExtractFromBroadcast()
1334 broadcastVecType.getShape().take_back(extractResultRank)) in foldExtractFromBroadcast()
1355 return type.getShape().take_back(n + 1).front(); in foldExtractFromShapeCast()
1609 SmallVector<int64_t, 4> newShape(type.getShape().begin(), in build()
1610 type.getShape().end()); in build()
1677 return llvm::to_vector<4>(getVectorType().getShape()); in getShapeForUnroll()
1828 llvm::append_range(shape, v1Type.getShape().drop_front()); in inferReturnTypes()
2207 auto sourceShape = sourceVectorType.getShape(); in verify()
2208 auto destShape = destVectorType.getShape(); in verify()
2418 if (fixedVectorSizes[i] != inputVectorType.getShape()[index]) in verify()
2426 if (fixedVectorSizes[i] != outputVectorType.getShape()[index]) in verify()
2472 for (unsigned e = vectorType.getShape().size(); idx < e; ++idx) in inferStridedSliceOpResultType()
2473 shape.push_back(vectorType.getShape()[idx]); in inferStridedSliceOpResultType()
2503 auto shape = type.getShape(); in verify()
2869 vectorElementType.getShape().back(); in verifyTransferOp()
2872 vectorType.getShape().back(); in verifyTransferOp()
2894 vectorType.getRank() == 0 ? 1 : vectorType.getShape().back(); in verifyTransferOp()
3185 return llvm::to_vector<4>(getVectorType().getShape()); in getShapeForUnroll()
3257 auto actualDestTensorShape = extractOp.getType().getShape(); in matchAndRewrite()
3260 inferredDestTensorType.getShape().take_back(vectorRank)) in matchAndRewrite()
3352 ArrayRef<int64_t> destShape = readOp.getVectorType().getShape(); in matchAndRewrite()
3464 auto maskType = VectorType::get(vectorType.getShape(), builder.getI1Type()); in parse()
3560 if (read.getVectorType().getShape() != rankedTensorType.getShape()) in foldReadInitWrite()
3623 return llvm::to_vector<4>(getVectorType().getShape()); in getShapeForUnroll()
3730 if (!llvm::equal(xferOp.getVectorType().getShape(), in matchAndRewrite()
3731 xferOp.getShapedType().getShape())) in matchAndRewrite()
3759 auto actualSourceTensorShape = insertOp.getSourceType().getShape(); in matchAndRewrite()
3762 inferredSourceTensorType.getShape().take_back(vectorRank)) in matchAndRewrite()
3848 ArrayRef<int64_t> vectorShape = transferOp.getVectorType().getShape(); in matchAndRewrite()
3850 transferOp.getPermutationMap(), transferOp.getShapedType().getShape()); in matchAndRewrite()
3858 transferOp.getPermutationMap(), insertOp.getSourceType().getShape()); in matchAndRewrite()
4288 auto sourceShape = sourceVectorType.getShape(); in verifyVectorShapeCast()
4289 auto resultShape = resultVectorType.getShape(); in verifyVectorShapeCast()
4337 if (!isValidShapeCast(srcType.getShape(), resultType.getShape())) in fold()
4340 if (!isValidShapeCast(resultType.getShape(), srcType.getShape())) in fold()
4401 ? broadcastSourceVectorType.getShape() in matchAndRewrite()
4403 auto shapeCastTargetShape = shapeCastOp.getResultVectorType().getShape(); in matchAndRewrite()
4448 } else if (sourceElementBits * sourceVectorType.getShape().back() != in verify()
4449 resultElementBits * resultVectorType.getShape().back()) { in verify()
4504 SmallVector<int64_t, 8> res(memRefType.getShape().begin(), in extractShape()
4505 memRefType.getShape().end()); in extractShape()
4507 res.append(vectorType.getShape().begin(), vectorType.getShape().end()); in extractShape()
4557 transposedShape[i] = vt.getShape()[transp[i]]; in build()
4611 return llvm::to_vector<4>(getResultType().getShape()); in getShapeForUnroll()
4730 auto resultShape = resultType.getShape(); in verify()
4807 createMaskOp.getType().getShape())) { in matchAndRewrite()
4855 ArrayRef<int64_t> srcShape = srcType.getShape(); in verify()
4856 ArrayRef<int64_t> initialValueShapes = initialType.getShape(); in verify()