`neural_compressor.experimental.data.filters.filter`

The base filter class for all frameworks.

Module Contents

`TensorflowFilters`	The base filter class for Tensorflow framework.
`ONNXRTQLFilters`	The base filter class for ONNXRT framework QLinear mode.
`ONNXRTITFilters`	The base filter class for ONNXRT framework IT mode.
`PyTorchFilters`	The base filter class for PyTorch framework.
`MXNetFilters`	The base filter class for MXNet framework.
`FILTERS`	The filter register for all frameworks.
`Filter`	The base class for transform.

filter_registry(filter_type, framework)

Register all transform subclasses.

class neural_compressor.experimental.data.filters.filter.TensorflowFilters[source]: The base filter class for Tensorflow framework.

class neural_compressor.experimental.data.filters.filter.ONNXRTQLFilters[source]: The base filter class for ONNXRT framework QLinear mode.

class neural_compressor.experimental.data.filters.filter.ONNXRTITFilters[source]: The base filter class for ONNXRT framework IT mode.

class neural_compressor.experimental.data.filters.filter.PyTorchFilters[source]: The base filter class for PyTorch framework.

class neural_compressor.experimental.data.filters.filter.MXNetFilters[source]: The base filter class for MXNet framework.

class neural_compressor.experimental.data.filters.filter.FILTERS(framework)[source]

The filter register for all frameworks.

Parameters:: framework (str) – frameworks in [“tensorflow”, “tensorflow_itex”, “mxnet”, “onnxrt_qdq”, “pytorch”, “pytorch_ipex”, “pytorch_fx”, “onnxrt_integerops”, “keras” “onnxrt_qlinearops”, “onnxruntime”].

neural_compressor.experimental.data.filters.filter.filter_registry(filter_type, framework)[source]

Register all transform subclasses.

Parameters:

filter_type (str) – fILTER registration name.
framework (str) – support 4 framework including ‘tensorflow’, ‘pytorch’, ‘mxnet’, ‘onnxrt’.
cls (class) – The class of register.

Returns:

The class of register.

Return type:

cls

class neural_compressor.experimental.data.filters.filter.Filter[source]

The base class for transform.

__call__ method is needed when write user specific transform.