I would definitely not use heat on the area that's swollen. It will draw more fluid to the area. Ice would be okay, though, if that's comfortable for you. Elevation and rest are good too, of course.
You're still fairly early in recovery, so be sure any massage that is done (you or the therapist) isn't overly aggressive. Gentle rubbing of the tissues can work well to help dissipate fluid. When I had my second TKR, I did my own lymphatic massage. With that, you do sweeping rubs from the base of the knee joint upwards over the thigh to your groin. The theory is the movement works fluid toward your body's core where your lymph system can more easily process it out through the kidneys. Lymph processing is notoriously slow, so this may speed things up a bit. If you can do this with no pain to the area that's swollen, you might give it a try. Of course, if it's sensitive to your touch, it's best to just leave it alone for now.