bitorch.quantizations.ste_heaviside

Sign Function Implementation

Classes

SteHeaviside()

Module for applying the SteHeaviside quantization, using an ste in backward pass

SteHeavisideFunction(*args, **kwargs)