Function callback

imperative/python/megengine/utils/module_stats.py:61–64

(impl)

def _register_dict(*modules, dict=None):
    def callback(impl):
        for module in modules:
            dict[module] = impl
        return impl

    return callback