EvoNorm layers in TensorFlow 2