Searched refs:rhsShape (Results 1 – 4 of 4) sorted by relevance
657 auto rhsShape = llvm::to_vector<6>( in fold() local663 if (!OpTrait::util::getBroadcastedShape(lhsShape, rhsShape, resultShape)) in fold()856 auto rhsShape = llvm::to_vector<6>( in fold() local860 resultShape.append(rhsShape.begin(), rhsShape.end()); in fold()
461 ShapeAdaptor rhsShape = operands.getShape(1); in inferReturnTypeComponents() local472 if (rhsShape.hasRank()) { in inferReturnTypeComponents()474 ? rhsShape.getDimSize(0) in inferReturnTypeComponents()476 outShape[2] = rhsShape.getDimSize(2); in inferReturnTypeComponents()
860 ArrayRef<int64_t> rhsShape, in findIntermediateShape() argument869 if (lhsShape.empty() || rhsShape.empty()) { in findIntermediateShape()875 while (currLhsDim < lhsShape.size() && currRhsDim < rhsShape.size()) { in findIntermediateShape()876 int64_t rhsSize = rhsShape[currRhsDim]; in findIntermediateShape()879 currRhsDim < rhsShape.size()) { in findIntermediateShape()885 rhsSize *= rhsShape[currRhsDim]; in findIntermediateShape()903 while (currRhsDim < rhsShape.size()) { in findIntermediateShape()904 if (rhsShape[currRhsDim++] != 1) { in findIntermediateShape()
614 ArrayRef<int64_t> rhsShape = contractOp.getRhsType().getShape(); in matchAndRewrite() local627 if (rhsShape[dim] != 1) in matchAndRewrite()678 rhsDims.append(rhsShape.begin(), rhsShape.end()); in matchAndRewrite()