`elasticai.creator.nn.quantized_grads.fixed_point.module_quantization`#

Module Contents#

`get_fxp_forw_quantization_module`
`get_fxp_forwbackw_quantization_module`

`QuantizeForwHTE`
`QuantizeForwStochastic`
`QuantizeForwHTEBackwHTE`
`QuantizeForwHTEBackwStochastic`
`QuantizeForwStochasticBackwStochastic`

elasticai.creator.nn.quantized_grads.fixed_point.module_quantization.get_fxp_forw_quantization_module(autograd_func: torch.autograd.Function) → type[elasticai.creator.nn.quantized_grads.fixed_point.module_quantization._ForwModule][source]#

elasticai.creator.nn.quantized_grads.fixed_point.module_quantization.get_fxp_forwbackw_quantization_module(autograd_func: torch.autograd.Function) → type[elasticai.creator.nn.quantized_grads.fixed_point.module_quantization._ForwBackwModule][source]#

elasticai.creator.nn.quantized_grads.fixed_point.module_quantization.QuantizeForwHTE: type[elasticai.creator.nn.quantized_grads.fixed_point.module_quantization._ForwModule]#: ‘get_fxp_forw_quantization_module(…)’

elasticai.creator.nn.quantized_grads.fixed_point.module_quantization.QuantizeForwStochastic: type[elasticai.creator.nn.quantized_grads.fixed_point.module_quantization._ForwModule]#: ‘get_fxp_forw_quantization_module(…)’

elasticai.creator.nn.quantized_grads.fixed_point.module_quantization.QuantizeForwHTEBackwHTE: type[elasticai.creator.nn.quantized_grads.fixed_point.module_quantization._ForwBackwModule]#: ‘get_fxp_forwbackw_quantization_module(…)’

elasticai.creator.nn.quantized_grads.fixed_point.module_quantization.QuantizeForwHTEBackwStochastic: type[elasticai.creator.nn.quantized_grads.fixed_point.module_quantization._ForwBackwModule]#: ‘get_fxp_forwbackw_quantization_module(…)’

elasticai.creator.nn.quantized_grads.fixed_point.module_quantization.QuantizeForwStochasticBackwStochastic: type[elasticai.creator.nn.quantized_grads.fixed_point.module_quantization._ForwBackwModule]#: ‘get_fxp_forwbackw_quantization_module(…)’