hgq.config package

Submodules

hgq.config.layer module

class hgq.config.layer.GlobalConfig

Bases: TypedDict

beta0: float

enable_ebops: bool

enable_iq: bool

enable_oq: bool

class hgq.config.layer.LayerConfigScope(*, beta0: float | None | Initializer = None, enable_ebops: bool | None = None)
class hgq.config.layer.LayerConfigScope(**kwargs): Bases: object

hgq.config.quantizer module

class hgq.config.quantizer.QuantizerConfig(q_type: str = 'kbi', place: str = 'datalane', *, k0: int | float | integer | floating | bool | Initializer = True, b0: int | float | integer | floating | Initializer = 4, i0: int | float | integer | floating | Initializer = 2, round_mode: str = 'RND', overflow_mode: str = 'WRAP', bc: Constraint | None = MinMax(0, 12), ic: Constraint | None = None, br: Regularizer | None = None, ir: Regularizer | None = None, i_decay_speed: int | float | integer | floating = -1, homogeneous_axis: Sequence[int] | None = None, heterogeneous_axis: Sequence[int] | None = None, bw_mapper: BitwidthMapperBase | None = None, scaler: int | float | integer | floating | None = None, qnoise_factor: float | None = None, **kwargs)

class hgq.config.quantizer.QuantizerConfig(q_type: str = 'kif', place: str = 'datalane', *, k0: int | float | integer | floating | bool | Initializer = True, i0: int | float | integer | floating | Initializer = 4, f0: int | float | integer | floating | Initializer = 2, round_mode: str = 'RND', overflow_mode: str = 'SAT', ic: Constraint | None = MinMax(-12, 12), ir: Regularizer | None = None, fc: Constraint | None = MinMax(-10, 10), fr: Regularizer | None = None, i_decay_speed: int | float | integer | floating = 0.01, homogeneous_axis: Sequence[int] | None = (0,), heterogeneous_axis: Sequence[int] | None = None, bw_mapper: BitwidthMapperBase | None = None, scaler: int | float | integer | floating | None = None, qnoise_factor: float | None = None, **kwargs)

class hgq.config.quantizer.QuantizerConfig(q_type: str = 'float', place: str = 'datalane', *, m0: int | float | integer | floating | Initializer = 2, e0: int | float | integer | floating | Initializer = 1, e00: int | float | integer | floating | Initializer = 0, mc: Constraint | None = Min(-1), ec: Constraint | None = MinMax(0, 4), e0c: Constraint | None = MinMax(-8, 8), mr: Regularizer | None = None, er: Regularizer | None = None, e0r: Regularizer | None = None, homogeneous_axis: Sequence[int] | None = (), heterogeneous_axis: Sequence[int] | None = None, bw_mapper: BitwidthMapperBase | None = None, scaler: int | float | integer | floating | None = None, qnoise_factor: float | None = None, **kwargs)

Bases: Mapping

classmethod from_config(config)

get_config()

get_quantizer() → TrainableQuantizerBase

class hgq.config.quantizer.QuantizerConfigScope(q_type: str | Sequence[str] | set[str] = 'all', place: str | Sequence[str] | set[str] = 'all', default_q_type=None, **kwargs)

Bases: object

override(): Override the default quantizer config.

hgq.config.quantizer.all_quantizer_types()

Module contents

class hgq.config.LayerConfigScope(*, beta0: float | None | Initializer = None, enable_ebops: bool | None = None)
class hgq.config.LayerConfigScope(**kwargs): Bases: object

class hgq.config.QuantizerConfig(q_type: str = 'kbi', place: str = 'datalane', *, k0: int | float | integer | floating | bool | Initializer = True, b0: int | float | integer | floating | Initializer = 4, i0: int | float | integer | floating | Initializer = 2, round_mode: str = 'RND', overflow_mode: str = 'WRAP', bc: Constraint | None = MinMax(0, 12), ic: Constraint | None = None, br: Regularizer | None = None, ir: Regularizer | None = None, i_decay_speed: int | float | integer | floating = -1, homogeneous_axis: Sequence[int] | None = None, heterogeneous_axis: Sequence[int] | None = None, bw_mapper: BitwidthMapperBase | None = None, scaler: int | float | integer | floating | None = None, qnoise_factor: float | None = None, **kwargs)

class hgq.config.QuantizerConfig(q_type: str = 'kif', place: str = 'datalane', *, k0: int | float | integer | floating | bool | Initializer = True, i0: int | float | integer | floating | Initializer = 4, f0: int | float | integer | floating | Initializer = 2, round_mode: str = 'RND', overflow_mode: str = 'SAT', ic: Constraint | None = MinMax(-12, 12), ir: Regularizer | None = None, fc: Constraint | None = MinMax(-10, 10), fr: Regularizer | None = None, i_decay_speed: int | float | integer | floating = 0.01, homogeneous_axis: Sequence[int] | None = (0,), heterogeneous_axis: Sequence[int] | None = None, bw_mapper: BitwidthMapperBase | None = None, scaler: int | float | integer | floating | None = None, qnoise_factor: float | None = None, **kwargs)

class hgq.config.QuantizerConfig(q_type: str = 'float', place: str = 'datalane', *, m0: int | float | integer | floating | Initializer = 2, e0: int | float | integer | floating | Initializer = 1, e00: int | float | integer | floating | Initializer = 0, mc: Constraint | None = Min(-1), ec: Constraint | None = MinMax(0, 4), e0c: Constraint | None = MinMax(-8, 8), mr: Regularizer | None = None, er: Regularizer | None = None, e0r: Regularizer | None = None, homogeneous_axis: Sequence[int] | None = (), heterogeneous_axis: Sequence[int] | None = None, bw_mapper: BitwidthMapperBase | None = None, scaler: int | float | integer | floating | None = None, qnoise_factor: float | None = None, **kwargs)

Bases: Mapping

classmethod from_config(config)

get_config()

get_quantizer() → TrainableQuantizerBase

class hgq.config.QuantizerConfigScope(q_type: str | Sequence[str] | set[str] = 'all', place: str | Sequence[str] | set[str] = 'all', default_q_type=None, **kwargs)

Bases: object

override(): Override the default quantizer config.