Searched defs:torch_dispatch_class (Results 1 – 2 of 2) sorted by relevance
78 def find(self, torch_dispatch_class): argument
187 def _register_torch_dispatch_rule(self, op_name, torch_dispatch_class, fn): argument