Fix order-dependent test caused by TF lazy module loading
Fix order-dependent test caused by TF lazy module loading

PiperOrigin-RevId: 299145425
Change-Id: I864708c875999126ee52e59313391796bf17a32b

tensorflow/gan,tensorflow/gan