Use tf.layers.utils instead of tf.contrib.layers.utils in batch_norm.
Use tf.layers.utils instead of tf.contrib.layers.utils in batch_norm.

PiperOrigin-RevId: 162344641

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