Corrected parameter list for testFusedBatchNorm, where `is_training` had been erroneously set to `True` in some cases.
Corrected parameter list for testFusedBatchNorm, where `is_training` had been erroneously set to `True` in some cases.

PiperOrigin-RevId: 179555601

rakshit-agrawal/sonnet,rakshit-agrawal/sonnet,deepmind/sonnet,deepmind/sonnet