elasticai.creator_plugins.quantized_grads

Subpackages

• elasticai.creator_plugins.quantized_grads.base_modules
  • elasticai.creator_plugins.quantized_grads.base_modules.conv1d
  • elasticai.creator_plugins.quantized_grads.base_modules.linear
  • elasticai.creator_plugins.quantized_grads.base_modules.batchnorm2d
  • elasticai.creator_plugins.quantized_grads.base_modules.conv2d
• elasticai.creator_plugins.quantized_grads.fixed_point
  • elasticai.creator_plugins.quantized_grads.fixed_point.autograd
  • elasticai.creator_plugins.quantized_grads.fixed_point.two_complement_fixed_point_config
  • elasticai.creator_plugins.quantized_grads.fixed_point.quantize_to_fixed_point
  • elasticai.creator_plugins.quantized_grads.fixed_point.param_quantization
  • elasticai.creator_plugins.quantized_grads.fixed_point.module_quantization
• elasticai.creator_plugins.quantized_grads.tests
  • elasticai.creator_plugins.quantized_grads.tests.fxpmodel_loss_optim_test
  • elasticai.creator_plugins.quantized_grads.tests.quantize_to_fixed_point_test
  • elasticai.creator_plugins.quantized_grads.tests.test_batchnorm
  • elasticai.creator_plugins.quantized_grads.tests.test_conv1d
  • elasticai.creator_plugins.quantized_grads.tests.two_complement_fixed_point_config_test
  • elasticai.creator_plugins.quantized_grads.tests.test_conv2d
  • elasticai.creator_plugins.quantized_grads.tests.test_linear

Submodules

• elasticai.creator_plugins.quantized_grads.identity_param_quantization
• elasticai.creator_plugins.quantized_grads.quantized_optim

Package Contents

Data

__all__

API

elasticai.creator_plugins.quantized_grads.__all__

[‘get_quantized_optimizer’, ‘Identity’]