diff --git a/model_compression_toolkit/core/common/quantization/quantization_config.py b/model_compression_toolkit/core/common/quantization/quantization_config.py index 50078fee9..b2da5bef6 100644 --- a/model_compression_toolkit/core/common/quantization/quantization_config.py +++ b/model_compression_toolkit/core/common/quantization/quantization_config.py @@ -24,7 +24,7 @@ class CustomOpsetLayers(NamedTuple): """ - This struct is used to define a set of operators to map to a custom operator set defined in the TPC. + This struct is used to define a set of operators to be used for setting a custom operator set in the TPC. Args: operators: a list of framework operators to map to a certain custom opset name. diff --git a/model_compression_toolkit/target_platform_capabilities/schema/v1.py b/model_compression_toolkit/target_platform_capabilities/schema/v1.py index a00d32953..d4df977d9 100644 --- a/model_compression_toolkit/target_platform_capabilities/schema/v1.py +++ b/model_compression_toolkit/target_platform_capabilities/schema/v1.py @@ -37,7 +37,7 @@ class OperatorSetNames(str, Enum): L2NORM = "L2Norm" RELU = "ReLU" RELU6 = "ReLU6" - LEAKY_RELU = "LEAKYReLU" + LEAKY_RELU = "LeakyReLU" ELU = "Elu" HARD_TANH = "HardTanh" ADD = "Add"