diff --git a/docs/source/quantization.rst b/docs/source/quantization.rst index 5bdd6948761..3be4d5a390e 100644 --- a/docs/source/quantization.rst +++ b/docs/source/quantization.rst @@ -1046,5 +1046,6 @@ Please take a look at `Limitations of Symbolic Tracing {**config} + if data_list is not None: + # add data with default config here + [self.add_data(name, data, **self.defaults) for name, data in data_list] + + def prepare(self): + raise NotImplementedError("this function is undefined for this class") + + def add_data(self, name: str, data, **config): + r""" Configures and parametrizes the internal container model with name and data + """ + pass + + def get_data(self, name: str): + r"""Returns weight tensor (or data) based on the input name. + """ + pass + + def __repr__(self): + r"""String representation of an object when printed + """ + pass + + def get_mask(self, name: str): + r"""Returns the mask currently associated with the named tensor. + """ + pass + + def squash_mask(self, *args, **kwargs): + r"""Squashes the sparse masks into the appropriate tensors. + """ + pass + + def step(self): + r"""Updates the mask for all the named data. + """ + pass + + @abc.abstractmethod + def update_mask(self, name, data, **kwargs): + pass