Roll P gain maybe too high. But yes you can set the TPA to just work on one, two, or all three axis. Or P,I, or D. I would try -.10% reduction on roll P in the TPA graph at the right(high) end and zero at the middle and left (low throttle) end of the graph
I had the high throttle shakes on one of my copters but couldn't decide if it was roll or pitch so I just did -10% and that smoothed out full throttle punches. I should probably tune it some, but I was in a hurry and anxious to fly it