elasticai.creator.nn.quantized_grads.base_modules

Submodules

• elasticai.creator.nn.quantized_grads.base_modules.conv1d
• elasticai.creator.nn.quantized_grads.base_modules.linear
• elasticai.creator.nn.quantized_grads.base_modules.batchnorm2d
• elasticai.creator.nn.quantized_grads.base_modules.conv2d

Package Contents

Data

__all__
parametrized_modules

API

elasticai.creator.nn.quantized_grads.base_modules.__all__

[‘ReLU’, ‘BatchNorm2d’, ‘Conv1d’, ‘Conv2d’, ‘Linear’, ‘parametrized_modules’]

elasticai.creator.nn.quantized_grads.base_modules.parametrized_modules

[‘ParametrizedBatchNorm2d’, ‘ParametrizedConv1d’, ‘ParametrizedConv2d’, ‘ParametrizedLinear’]