Function canonicalize_arg

imperative/python/megengine/xla/dtype.py:104–109

(x)

def canonicalize_arg(x):
    typ = type(x)
    handler = canonicalize_args_handlers.get(typ)
    if handler:
        return handler(x)
    raise TypeError(f"No canonicalize_dtype handler for type: {type(x)}")