diff --git a/ai_edge_quantizer/algorithm_manager.py b/ai_edge_quantizer/algorithm_manager.py index d1a120e1..d36390e2 100644 --- a/ai_edge_quantizer/algorithm_manager.py +++ b/ai_edge_quantizer/algorithm_manager.py @@ -64,6 +64,10 @@ class AlgorithmName(str, enum.Enum): HADAMARD_ROTATION = hadamard_rotation.ALGORITHM_KEY MSE = mse.ALGORITHM_KEY + @classmethod + def list(cls) -> list[str]: + return list(map(lambda c: c.value, cls)) + ### MIN/MAX_UNIFORM_QUANT ###