bitorch.layers.registerΒΆ

Functions

all_layer_registries

Return all layer registries (one for each layer type: QLinear, QConv[1-3]d).

convert_layers_to

Convert all wrapped layers (or a given subset of them) to a new mode.

Classes

QConv1dImplementation(supports_modes)

Decorator for QConv1d implementations.

QConv2dImplementation(supports_modes)

Decorator for QConv2d implementations.

QConv3dImplementation(supports_modes)

Decorator for QConv3d implementations.

QLinearImplementation(supports_modes)

Decorator for QLinear implementations.