`elasticai.creator.nn.fixed_point.round_to_fixed_point`#

Module Contents#

Classes#

`RoundToFixedPoint`
`CutToFixedPoint`

API#

class elasticai.creator.nn.fixed_point.round_to_fixed_point.RoundToFixedPoint(*args, **kwargs)[source]#

Bases: torch.autograd.Function

static forward(ctx: Any, *args: Any, **kwargs: Any) → torch.Tensor[source]#

static backward(ctx: Any, *grad_outputs: Any) → Any[source]#

class elasticai.creator.nn.fixed_point.round_to_fixed_point.CutToFixedPoint(*args, **kwargs)[source]#

Bases: torch.autograd.Function

static forward(ctx: Any, *args: Any, **kwargs: Any) → torch.Tensor[source]#

static backward(ctx: Any, *grad_outputs: Any) → Any[source]#