bitorch.layersΒΆ
This submodule contains adapted pytorch layers that use quantization functions on their weights and activations before forwarding them. These layers use the quantization functions specified in the quantization submodule.
Functions
Convert the given module to a new bitorch RuntimeMode. |
Modules
Config class for quantization layers. |
|
This submodule contains objects needed to provide and manage custom layer implementations. |
|
Module containing the quantized 1d convolution layer |
|
Module containing the quantized 2d convolution layer |
|
Module containing the quantized 3d convolution layer |
|
Module containing the quantized linear layer |
|