RMSNormalization
================

.. currentmodule:: lib.model.normalization

.. autoclass:: RMSNormalization
   :show-inheritance:
   :noindex:

   .. rubric:: Attributes Summary

   .. autosummary::

      ~RMSNormalization.T_destination
      ~RMSNormalization.call_super_init
      ~RMSNormalization.compute_dtype
      ~RMSNormalization.dtype
      ~RMSNormalization.dtype_policy
      ~RMSNormalization.dump_patches
      ~RMSNormalization.input
      ~RMSNormalization.input_dtype
      ~RMSNormalization.input_spec
      ~RMSNormalization.losses
      ~RMSNormalization.metrics
      ~RMSNormalization.metrics_variables
      ~RMSNormalization.non_trainable_variables
      ~RMSNormalization.non_trainable_weights
      ~RMSNormalization.output
      ~RMSNormalization.path
      ~RMSNormalization.quantization_mode
      ~RMSNormalization.supports_masking
      ~RMSNormalization.torch_params
      ~RMSNormalization.trainable
      ~RMSNormalization.trainable_variables
      ~RMSNormalization.trainable_weights
      ~RMSNormalization.variable_dtype
      ~RMSNormalization.variables
      ~RMSNormalization.weights

   .. rubric:: Methods Summary

   .. autosummary::

      ~RMSNormalization.__call__
      ~RMSNormalization.add_loss
      ~RMSNormalization.add_metric
      ~RMSNormalization.add_module
      ~RMSNormalization.add_variable
      ~RMSNormalization.add_weight
      ~RMSNormalization.apply
      ~RMSNormalization.bfloat16
      ~RMSNormalization.buffers
      ~RMSNormalization.build
      ~RMSNormalization.build_from_config
      ~RMSNormalization.call
      ~RMSNormalization.children
      ~RMSNormalization.compile
      ~RMSNormalization.compute_mask
      ~RMSNormalization.compute_output_shape
      ~RMSNormalization.compute_output_spec
      ~RMSNormalization.count_params
      ~RMSNormalization.cpu
      ~RMSNormalization.cuda
      ~RMSNormalization.double
      ~RMSNormalization.eval
      ~RMSNormalization.extra_repr
      ~RMSNormalization.float
      ~RMSNormalization.forward
      ~RMSNormalization.from_config
      ~RMSNormalization.get_buffer
      ~RMSNormalization.get_build_config
      ~RMSNormalization.get_config
      ~RMSNormalization.get_extra_state
      ~RMSNormalization.get_parameter
      ~RMSNormalization.get_submodule
      ~RMSNormalization.get_weights
      ~RMSNormalization.half
      ~RMSNormalization.ipu
      ~RMSNormalization.load_own_variables
      ~RMSNormalization.load_state_dict
      ~RMSNormalization.modules
      ~RMSNormalization.mtia
      ~RMSNormalization.named_buffers
      ~RMSNormalization.named_children
      ~RMSNormalization.named_modules
      ~RMSNormalization.named_parameters
      ~RMSNormalization.parameters
      ~RMSNormalization.quantize
      ~RMSNormalization.quantized_build
      ~RMSNormalization.quantized_call
      ~RMSNormalization.register_backward_hook
      ~RMSNormalization.register_buffer
      ~RMSNormalization.register_forward_hook
      ~RMSNormalization.register_forward_pre_hook
      ~RMSNormalization.register_full_backward_hook
      ~RMSNormalization.register_full_backward_pre_hook
      ~RMSNormalization.register_load_state_dict_post_hook
      ~RMSNormalization.register_load_state_dict_pre_hook
      ~RMSNormalization.register_module
      ~RMSNormalization.register_parameter
      ~RMSNormalization.register_state_dict_post_hook
      ~RMSNormalization.register_state_dict_pre_hook
      ~RMSNormalization.rematerialized_call
      ~RMSNormalization.requires_grad_
      ~RMSNormalization.save_own_variables
      ~RMSNormalization.set_extra_state
      ~RMSNormalization.set_submodule
      ~RMSNormalization.set_weights
      ~RMSNormalization.share_memory
      ~RMSNormalization.state_dict
      ~RMSNormalization.stateless_call
      ~RMSNormalization.symbolic_call
      ~RMSNormalization.to
      ~RMSNormalization.to_empty
      ~RMSNormalization.train
      ~RMSNormalization.type
      ~RMSNormalization.xpu
      ~RMSNormalization.zero_grad

   .. rubric:: Attributes Documentation

   .. autoattribute:: T_destination
   .. autoattribute:: call_super_init
   .. autoattribute:: compute_dtype
   .. autoattribute:: dtype
   .. autoattribute:: dtype_policy
   .. autoattribute:: dump_patches
   .. autoattribute:: input
   .. autoattribute:: input_dtype
   .. autoattribute:: input_spec
   .. autoattribute:: losses
   .. autoattribute:: metrics
   .. autoattribute:: metrics_variables
   .. autoattribute:: non_trainable_variables
   .. autoattribute:: non_trainable_weights
   .. autoattribute:: output
   .. autoattribute:: path
   .. autoattribute:: quantization_mode
   .. autoattribute:: supports_masking
   .. autoattribute:: torch_params
   .. autoattribute:: trainable
   .. autoattribute:: trainable_variables
   .. autoattribute:: trainable_weights
   .. autoattribute:: variable_dtype
   .. autoattribute:: variables
   .. autoattribute:: weights

   .. rubric:: Methods Documentation

   .. automethod:: __call__
   .. automethod:: add_loss
   .. automethod:: add_metric
   .. automethod:: add_module
   .. automethod:: add_variable
   .. automethod:: add_weight
   .. automethod:: apply
   .. automethod:: bfloat16
   .. automethod:: buffers
   .. automethod:: build
   .. automethod:: build_from_config
   .. automethod:: call
   .. automethod:: children
   .. automethod:: compile
   .. automethod:: compute_mask
   .. automethod:: compute_output_shape
   .. automethod:: compute_output_spec
   .. automethod:: count_params
   .. automethod:: cpu
   .. automethod:: cuda
   .. automethod:: double
   .. automethod:: eval
   .. automethod:: extra_repr
   .. automethod:: float
   .. automethod:: forward
   .. automethod:: from_config
   .. automethod:: get_buffer
   .. automethod:: get_build_config
   .. automethod:: get_config
   .. automethod:: get_extra_state
   .. automethod:: get_parameter
   .. automethod:: get_submodule
   .. automethod:: get_weights
   .. automethod:: half
   .. automethod:: ipu
   .. automethod:: load_own_variables
   .. automethod:: load_state_dict
   .. automethod:: modules
   .. automethod:: mtia
   .. automethod:: named_buffers
   .. automethod:: named_children
   .. automethod:: named_modules
   .. automethod:: named_parameters
   .. automethod:: parameters
   .. automethod:: quantize
   .. automethod:: quantized_build
   .. automethod:: quantized_call
   .. automethod:: register_backward_hook
   .. automethod:: register_buffer
   .. automethod:: register_forward_hook
   .. automethod:: register_forward_pre_hook
   .. automethod:: register_full_backward_hook
   .. automethod:: register_full_backward_pre_hook
   .. automethod:: register_load_state_dict_post_hook
   .. automethod:: register_load_state_dict_pre_hook
   .. automethod:: register_module
   .. automethod:: register_parameter
   .. automethod:: register_state_dict_post_hook
   .. automethod:: register_state_dict_pre_hook
   .. automethod:: rematerialized_call
   .. automethod:: requires_grad_
   .. automethod:: save_own_variables
   .. automethod:: set_extra_state
   .. automethod:: set_submodule
   .. automethod:: set_weights
   .. automethod:: share_memory
   .. automethod:: state_dict
   .. automethod:: stateless_call
   .. automethod:: symbolic_call
   .. automethod:: to
   .. automethod:: to_empty
   .. automethod:: train
   .. automethod:: type
   .. automethod:: xpu
   .. automethod:: zero_grad