Searched refs:lhsShape (Results 1 – 5 of 5) sorted by relevance
655 auto lhsShape = llvm::to_vector<6>( in fold() local663 if (!OpTrait::util::getBroadcastedShape(lhsShape, rhsShape, resultShape)) in fold()854 auto lhsShape = llvm::to_vector<6>( in fold() local859 resultShape.append(lhsShape.begin(), lhsShape.end()); in fold()
460 ShapeAdaptor lhsShape = operands.getShape(0); in inferReturnTypeComponents() local467 if (lhsShape.hasRank()) { in inferReturnTypeComponents()468 outShape[0] = lhsShape.getDimSize(0); in inferReturnTypeComponents()469 outShape[1] = lhsShape.getDimSize(1); in inferReturnTypeComponents()
859 static bool findIntermediateShape(ArrayRef<int64_t> lhsShape, in findIntermediateShape() argument869 if (lhsShape.empty() || rhsShape.empty()) { in findIntermediateShape()875 while (currLhsDim < lhsShape.size() && currRhsDim < rhsShape.size()) { in findIntermediateShape()877 int64_t lhsSize = lhsShape[currLhsDim]; in findIntermediateShape()878 while (lhsSize != rhsSize && currLhsDim < lhsShape.size() && in findIntermediateShape()882 lhsSize *= lhsShape[currLhsDim]; in findIntermediateShape()897 while (currLhsDim < lhsShape.size()) { in findIntermediateShape()898 if (lhsShape[currLhsDim++] != 1) { in findIntermediateShape()
613 ArrayRef<int64_t> lhsShape = contractOp.getLhsType().getShape(); in matchAndRewrite() local623 if (lhsShape[dim] != 1) in matchAndRewrite()672 lhsDims.append(lhsShape.begin(), lhsShape.end()); in matchAndRewrite()
834 auto lhsShape = getLhsType().getShape(); in getIterationBounds() local846 iterationBounds.push_back(lhsShape[lhsDimIndex]); in getIterationBounds()