Two studies have shown that essential amino acids (EAA) can help function, and suppress atrophy of the rectus after TKA.
Dreyer et al. J Clinc Invest. 2013;123(11):4654-4666. Essential amino acid supplementation in patients following total knee arthroplasty.
Ueyama et al. The Bone & Joint Journal Vol 102-B, No. 6, Supp A. Perioperative essential amino acid supplementation suppresses rectus femoris muscle atrophy and accelerates early functional recovery following total knee arthroplasty.
The two brands I recommend to patients are Thorne ( https://amzn.to/3KPuC2i ) and Pure Encapsulations ( https://amzn.to/3ObJj1U )