docs: def _parse_normalization_kwargs(self, use_batch_norm, batch_norm_config,
                                    normalization_ctor, normalization_kwargs):
      """Sets up normalization, checking old and new flags."""
      if use_batch_norm is not None:
        # Delete this whole block when deprecation is done.
        util.deprecation_warning(
            "`use_batch_norm` kwarg is deprecated. Change your code to instead "
            "specify `normalization_ctor` and `normalization_kwargs`.")
        if not use_batch_norm:
          # Explicitly set to False - normalization_{ctor,kwargs} has precedence.
          self._check_and_assign_normalization_members(normalization_ctor,
                                                       normalization_kwargs or {})
        else:  # Explicitly set to true - new kwargs must not be used.
          if normalization_ctor is not None or normalization_kwargs is not None:
            raise ValueError(
                "if use_batch_norm is specified, normalization_ctor and "
                "normalization_kwargs must not be.")
          self._check_and_assign_normalization_members(batch_norm.BatchNorm,
                                                       batch_norm_config or {})
      else:
        # Old kwargs not set, this block will remain after removing old kwarg.
        self._check_and_assign_normalization_members(normalization_ctor,
                                                     normalization_kwargs or {})