
    3j                     l    S SK r S SKJs  Jr  SS.S\ R
                  S\S\S\S\ R
                  4
S	 jjr	g)
    Ng        )valuetensormultiplemoder   returnc                    U R                  5       u    pEnXU-  -
  U-  nXU-  -
  U-  nU(       d  U(       a  [        R                  " U SUSU4X#5      $ U $ )a7  
Pad a tensor's size to a multiple of a number.

Args:
    tensor: Tensor to pad.
    multiple: Size multiple to pad to.
    mode: Padding mode; see `torch.nn.functional.pad`.
    value: Padding value; see `torch.nn.functional.pad`.

Returns:
    Padded tensor, or the original tensor if no padding was needed.
r   )sizeFpad)	r   r   r   r   _hwpad_hpad_ws	            g/home/wildlama/miniconda3/lib/python3.13/site-packages/spandrel/architectures/__arch_helpers/padding.pypad_to_multipler      s]    & JAqQH$0EH$0EuuVa514??M    )
torchtorch.nn.functionalnn
functionalr
   Tensorintstrfloatr    r   r   <module>r      sN       LL 	
  \\r   