Lines Matching defs:resultShape

611       llvm::ArrayRef<int64_t> resultShape = resultExpr.getShape();
613       if (resultShape.size() != (maskShape.size() - 1))
677       llvm::ArrayRef<int64_t> resultShape = resultExpr.getShape();
679       if (resultShape.size() != (maskShape.size() - 1))
831       llvm::ArrayRef<int64_t> resultShape = resultExpr.getShape();
833       if (resultShape.size() != (arrayShape.size() - 1))
896     llvm::ArrayRef<int64_t> resultShape = resultExpr.getShape();
898     if (resultShape.size() != (arrayShape.size() - 1))
984     llvm::ArrayRef<int64_t> resultShape = resultExpr.getShape();
986     if (dim && resultShape.size() != (arrayShape.size() - 1))
990     if (!dim && resultShape.size() != 1)
1146   llvm::ArrayRef<int64_t> resultShape = resultTy.getShape();
1189   if (resultShape.size() != expectedResultShape.size())
1191   if (resultShape[0] != expectedResultShape[0] &&
1194   if (resultShape.size() == 2 && resultShape[1] != expectedResultShape[1] &&
1269   llvm::ArrayRef<int64_t> resultShape = resultTy.getShape();
1270   std::size_t resultRank = resultShape.size();
1280   if ((inShape[0] != resultShape[1]) && (inShape[0] != unknownExtent))
1282   if ((inShape[1] != resultShape[0]) && (inShape[1] != unknownExtent))
1317   llvm::ArrayRef<int64_t> resultShape = resultTy.getShape();
1355   if (resultShape.size() != expectedResultShape.size())
1357   if (resultShape[0] != expectedResultShape[0])
1359   if (resultShape.size() == 2 && resultShape[1] != expectedResultShape[1])
1384   llvm::ArrayRef<int64_t> resultShape = resultTy.getShape();
1385   std::size_t resultRank = resultShape.size();
1401   for (auto [inDim, resultDim] : llvm::zip(inShape, resultShape))