elasticai.creator.nn.quantized_grads.fixed_point.autograd

Module Contents

Functions

get_autograd_func

API

elasticai.creator.nn.quantized_grads.fixed_point.autograd.get_autograd_func(forw_func: Callable[[torch.Tensor, torch.Tensor, torch.Tensor, torch.Tensor], torch.Tensor], backw_func: Callable[[torch.Tensor, torch.Tensor, torch.Tensor, torch.Tensor], torch.Tensor]) → tuple[type[torch.autograd.Function], type[torch.autograd.Function]]