PixelShuffler
=============

.. currentmodule:: lib.model.layers

.. autoclass:: PixelShuffler
   :show-inheritance:
   :noindex:

   .. rubric:: Attributes Summary

   .. autosummary::

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

   .. rubric:: Methods Summary

   .. autosummary::

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