def get_palettizer_module(cls, module: _nn.Module) -> _Optional[_Type[_nn.Module]]:
        """
        Return :py:class:`DKMPalettizerModule` corresponding to given module.
        """
        if type(module) in cls.REGISTRY:
            return cls.REGISTRY[type(module)]
        return None

        Return :py:class:`DKMPalettizerModule` corresponding to given module.
        
coremltools/optimize/torch/palettization/_supported_modules.py