code: def _get_maybe_norm_layer(self, model):
      if model.name == "maybe_norm_layer":
          return model
      else:
          return model.get_layer("maybe_norm_layer")
  
imports: import tempfile
  from absl.testing import parameterized
  import numpy
  import tensorflow.compat.v2
  from tensorflow_probability.python.internal import test_util
  from tensorflow_probability.python.internal import tf_keras
  from tensorflow_probability.python.layers import weight_norm
label_desc: Use image node_id 3 for calling the WeightNormTest obj's underlying member method code with example usage: obj._get_maybe_norm_layer(model) and returns: model, model