75     def __init__(self, ns, kind, dispatch_key=""):  argument
88             kind, ns, dispatch_key, filename, lineno
95         self.dispatch_key = dispatch_key
208     def _impl_with_aoti_compile(self, op_name, dispatch_key=""):  argument
220         if dispatch_key == "":
221             dispatch_key = self.dispatch_key
222         assert torch.DispatchKeySet(dispatch_key).has(torch._C.DispatchKey.Dense)
237         key = self.ns + "/" + name.split("::")[-1] + "/" + dispatch_key
244                     name.split("::")[-1], dispatch_key, self.ns
250         impl_fn(self.ns, name.split("::")[-1], dispatch_key)
255     def impl(self, op_name, fn, dispatch_key="", *, with_keyset=False):  argument
277         if dispatch_key == "":
278             dispatch_key = self.dispatch_key
292         key = self.ns + "/" + name.split("::")[-1] + "/" + dispatch_key
299                     name.split("::")[-1], dispatch_key, self.ns
303         if dispatch_key == "Meta":
325             dispatch_key if dispatch_key != "" else "CompositeImplicitAutograd",
333     def fallback(self, fn, dispatch_key="", *, with_keyset=False):  argument
353         if dispatch_key == "":
354             dispatch_key = self.dispatch_key
361         assert dispatch_key != ""
364         self.m.fallback(dispatch_key, fn, with_keyset)
591 def _(lib: Library, name, dispatch_key=""):  argument
595         lib.impl(name, f, dispatch_key)