Index

_

__init__() (tiny_diff.losses.discriminator.d_loss.DLoss method) (tiny_diff.losses.discriminator.hinge.HingeDLoss method) (tiny_diff.losses.discriminator.vanilla.VanillaDLoss method) (tiny_diff.losses.vae_loss.BetaPolicy method) (tiny_diff.losses.vae_loss.VAELoss method) (tiny_diff.models.conv_model.ConvModel method) (tiny_diff.models.discriminator.Pix2PixDiscriminator method) (tiny_diff.models.unet.layers.DownUNetLayer method) (tiny_diff.models.unet.layers.TimeEmbedding method) (tiny_diff.models.unet.layers.UNetLayer method) (tiny_diff.models.unet.layers.UNetLayerMixin method) (tiny_diff.models.unet.layers.UNetSkipLayer method) (tiny_diff.models.unet.layers.UNetSkipLayerMixin method) (tiny_diff.models.unet.layers.UpUNetLayer method) (tiny_diff.models.unet.model.UNet method) (tiny_diff.models.vae.components.ConvDecoder method) (tiny_diff.models.vae.components.ConvEncoder method) (tiny_diff.models.vae.components.ConvVAEComponent method) (tiny_diff.models.vae.gan.VAEGAN method) (tiny_diff.models.vae.gan.VAEGANAdapt method) (tiny_diff.models.vae.layers.LatentAlignLayer method) (tiny_diff.models.vae.model.ConvVAE method) (tiny_diff.models.vae.model.VAE method) (tiny_diff.modules.attention.AttnFlatten method) (tiny_diff.modules.attention.CrossVisualAttention method) (tiny_diff.modules.attention.SelfVisualAttention method) (tiny_diff.modules.attention.VisualAttention method) (tiny_diff.modules.conv.ConvBlock method) (tiny_diff.modules.conv.PreNormActConvBlock method) (tiny_diff.modules.conv.UpsampleConvBlock method) (tiny_diff.modules.layer_factory.LayerFactory method) (tiny_diff.modules.nonlinearity.Swish method) (tiny_diff.modules.norm.GroupNorm method) (tiny_diff.modules.resampling.Downsample method) (tiny_diff.modules.resampling.Interpolation method) (tiny_diff.modules.resampling.Upsample method) (tiny_diff.modules.residual.blocks.ERABlock method) (tiny_diff.modules.residual.blocks.ERARBlock method) (tiny_diff.modules.residual.blocks.ERBlock method) (tiny_diff.modules.residual.blocks.RABlock method) (tiny_diff.modules.residual.blocks.RABlockABC method) (tiny_diff.modules.residual.blocks.RBlock method) (tiny_diff.modules.residual.layers.ConvResizeResidualLayerABC method) (tiny_diff.modules.residual.layers.DownsampleResidualLayer method) (tiny_diff.modules.residual.layers.EResidualLayer method) (tiny_diff.modules.residual.layers.InterpolationResidualLayerABC method) (tiny_diff.modules.residual.layers.ResidualLayer method) (tiny_diff.modules.residual.layers.ResidualLayerABC method) (tiny_diff.modules.residual.layers.ResizeResidualLayerABC method) (tiny_diff.modules.residual.layers.UpsampleResidualLayer method) (tiny_diff.schedulers.ddpm.DDPMScheduler method) (tiny_diff.schedulers.ddpm.LatentDDPMScheduler method)

A

AttnFlatten (class in tiny_diff.modules.attention)

B

BetaPolicy (class in tiny_diff.losses.vae_loss)

C

ConvBlock (class in tiny_diff.modules.conv) ConvDecoder (class in tiny_diff.models.vae.components) ConvEncoder (class in tiny_diff.models.vae.components) ConvModel (class in tiny_diff.models.conv_model)

ConvResizeResidualLayerABC (class in tiny_diff.modules.residual.layers) ConvVAE (class in tiny_diff.models.vae.model) ConvVAEComponent (class in tiny_diff.models.vae.components) CrossVisualAttention (class in tiny_diff.modules.attention)

D

DDPMScheduler (class in tiny_diff.schedulers.ddpm) DLoss (class in tiny_diff.losses.discriminator.d_loss)

Downsample (class in tiny_diff.modules.resampling) DownsampleResidualLayer (class in tiny_diff.modules.residual.layers) DownUNetLayer (class in tiny_diff.models.unet.layers)

E

ERABlock (class in tiny_diff.modules.residual.blocks) ERARBlock (class in tiny_diff.modules.residual.blocks)

ERBlock (class in tiny_diff.modules.residual.blocks) EResidualLayer (class in tiny_diff.modules.residual.layers)

G

GroupNorm (class in tiny_diff.modules.norm)

H

HingeDLoss (class in tiny_diff.losses.discriminator.hinge)

I

Interpolation (class in tiny_diff.modules.resampling)

InterpolationResidualLayerABC (class in tiny_diff.modules.residual.layers)

L

LatentAlignLayer (class in tiny_diff.models.vae.layers) LatentDDPMScheduler (class in tiny_diff.schedulers.ddpm)

LayerFactory (class in tiny_diff.modules.layer_factory) loss_backward() (in module tiny_diff.utils)

M

make_fwd_ignore_context() (in module tiny_diff.models.unet.wrappers) match_shape() (in module tiny_diff.utils) module tiny_diff tiny_diff.constants tiny_diff.losses tiny_diff.losses.discriminator tiny_diff.losses.discriminator.d_loss tiny_diff.losses.discriminator.hinge tiny_diff.losses.discriminator.vanilla tiny_diff.losses.normal tiny_diff.losses.vae_loss tiny_diff.models tiny_diff.models.conv_model tiny_diff.models.discriminator tiny_diff.models.unet tiny_diff.models.unet.layers tiny_diff.models.unet.model tiny_diff.models.unet.wrappers tiny_diff.models.vae tiny_diff.models.vae.components tiny_diff.models.vae.gan tiny_diff.models.vae.layers tiny_diff.models.vae.model tiny_diff.modules tiny_diff.modules.attention tiny_diff.modules.conv tiny_diff.modules.layer_factory tiny_diff.modules.nonlinearity tiny_diff.modules.norm tiny_diff.modules.resampling tiny_diff.modules.residual tiny_diff.modules.residual.blocks tiny_diff.modules.residual.layers tiny_diff.schedulers tiny_diff.schedulers.ddpm tiny_diff.utils

mv_normal_log_likelihood() (in module tiny_diff.losses.normal)

P

Pix2PixDiscriminator (class in tiny_diff.models.discriminator)

PreNormActConvBlock (class in tiny_diff.modules.conv)

R

RABlock (class in tiny_diff.modules.residual.blocks) RABlockABC (class in tiny_diff.modules.residual.blocks) RBlock (class in tiny_diff.modules.residual.blocks)

ResidualLayer (class in tiny_diff.modules.residual.layers) ResidualLayerABC (class in tiny_diff.modules.residual.layers) ResizeResidualLayerABC (class in tiny_diff.modules.residual.layers)

S

SelfVisualAttention (class in tiny_diff.modules.attention)

Swish (class in tiny_diff.modules.nonlinearity)

T

TimeEmbedding (class in tiny_diff.models.unet.layers) tiny_diff module tiny_diff.constants module tiny_diff.losses module tiny_diff.losses.discriminator module tiny_diff.losses.discriminator.d_loss module tiny_diff.losses.discriminator.hinge module tiny_diff.losses.discriminator.vanilla module tiny_diff.losses.normal module tiny_diff.losses.vae_loss module tiny_diff.models module tiny_diff.models.conv_model module tiny_diff.models.discriminator module tiny_diff.models.unet module tiny_diff.models.unet.layers module tiny_diff.models.unet.model module tiny_diff.models.unet.wrappers module tiny_diff.models.vae module

tiny_diff.models.vae.components module tiny_diff.models.vae.gan module tiny_diff.models.vae.layers module tiny_diff.models.vae.model module tiny_diff.modules module tiny_diff.modules.attention module tiny_diff.modules.conv module tiny_diff.modules.layer_factory module tiny_diff.modules.nonlinearity module tiny_diff.modules.norm module tiny_diff.modules.resampling module tiny_diff.modules.residual module tiny_diff.modules.residual.blocks module tiny_diff.modules.residual.layers module tiny_diff.schedulers module tiny_diff.schedulers.ddpm module tiny_diff.utils module

U

UNet (class in tiny_diff.models.unet.model) UNetLayer (class in tiny_diff.models.unet.layers) UNetLayerMixin (class in tiny_diff.models.unet.layers) UNetSkipLayer (class in tiny_diff.models.unet.layers)

UNetSkipLayerMixin (class in tiny_diff.models.unet.layers) Upsample (class in tiny_diff.modules.resampling) UpsampleConvBlock (class in tiny_diff.modules.conv) UpsampleResidualLayer (class in tiny_diff.modules.residual.layers) UpUNetLayer (class in tiny_diff.models.unet.layers)

V

VAE (class in tiny_diff.models.vae.model) VAEGAN (class in tiny_diff.models.vae.gan) VAEGANAdapt (class in tiny_diff.models.vae.gan)

VAELoss (class in tiny_diff.losses.vae_loss) VanillaDLoss (class in tiny_diff.losses.discriminator.vanilla) VisualAttention (class in tiny_diff.modules.attention)

Z

zero_module() (in module tiny_diff.modules.conv)