I guess so. Never done that myself.
Couldn't do it in twrp it said there was no such path for /dev/block/bootdevice/by-name/TA. I decided to flash stock first then root it and restore TA from Android. That worked fine just had to change "adb shell" command to "adb shell su -c" or I would get permission denied error.