elasticai.creator.nn.quantized_grads#

Package Contents#

Classes#

Identity

Functions#

get_quantized_optimizer

API#

elasticai.creator.nn.quantized_grads.get_quantized_optimizer(optimizer: type[torch.optim.Optimizer]) → type[elasticai.creator.nn.quantized_grads.quantized_optim._QOptim][source]#
class elasticai.creator.nn.quantized_grads.Identity(*args: Any, **kwargs: Any)[source]#

Bases: torch.nn.Identity

static right_inverse(x: torch.Tensor) → torch.Tensor[source]#