tiny\_diff.models.unet.layers.UNetLayer
=======================================

.. currentmodule:: tiny_diff.models.unet.layers

.. autoclass:: UNetLayer
   :members:                                    <-- add at least this line
   :show-inheritance:                           <-- plus I want to show inheritance...
   :no-inherited-members:                          <-- ...and inherited members too

   
   .. automethod:: __init__

   
   .. rubric:: Methods

   .. autosummary::
   
      ~UNetLayer.__init__
      ~UNetLayer.add_module
      ~UNetLayer.apply
      ~UNetLayer.bfloat16
      ~UNetLayer.buffers
      ~UNetLayer.children
      ~UNetLayer.compile
      ~UNetLayer.cpu
      ~UNetLayer.cuda
      ~UNetLayer.double
      ~UNetLayer.eval
      ~UNetLayer.extra_repr
      ~UNetLayer.float
      ~UNetLayer.forward
      ~UNetLayer.get_buffer
      ~UNetLayer.get_extra_state
      ~UNetLayer.get_parameter
      ~UNetLayer.get_submodule
      ~UNetLayer.half
      ~UNetLayer.ipu
      ~UNetLayer.load_state_dict
      ~UNetLayer.modules
      ~UNetLayer.named_buffers
      ~UNetLayer.named_children
      ~UNetLayer.named_modules
      ~UNetLayer.named_parameters
      ~UNetLayer.parameters
      ~UNetLayer.register_backward_hook
      ~UNetLayer.register_buffer
      ~UNetLayer.register_forward_hook
      ~UNetLayer.register_forward_pre_hook
      ~UNetLayer.register_full_backward_hook
      ~UNetLayer.register_full_backward_pre_hook
      ~UNetLayer.register_load_state_dict_post_hook
      ~UNetLayer.register_module
      ~UNetLayer.register_parameter
      ~UNetLayer.register_state_dict_pre_hook
      ~UNetLayer.requires_grad_
      ~UNetLayer.residual_forward
      ~UNetLayer.set_extra_state
      ~UNetLayer.share_memory
      ~UNetLayer.state_dict
      ~UNetLayer.to
      ~UNetLayer.to_empty
      ~UNetLayer.train
      ~UNetLayer.type
      ~UNetLayer.xpu
      ~UNetLayer.zero_grad
   
   

   
   
   .. rubric:: Attributes

   .. autosummary::
   
      ~UNetLayer.T_destination
      ~UNetLayer.attention
      ~UNetLayer.call_super_init
      ~UNetLayer.dump_patches
      ~UNetLayer.last_layer
      ~UNetLayer.residual_block_cls
      ~UNetLayer.training