MCPcopy Create free account
hub / github.com/pytorch/pytorch / registrations

Method registrations

torch/_python_dispatcher.py:154–163  ·  view source on GitHub ↗
(self)

Source from the content-addressed store, hash-verified

152 """
153
154 def registrations(self):
155 output = self._format_header("Registered Kernels")
156 state = self.rawRegistrations()
157 state_entries = state.split("\n")
158 for line in state_entries:
159 first = line.split(":")[0]
160 if any(first.startswith(k) for k in self.supported_keys):
161 kernel = line.split("::")[0].split(" ")[1]
162 output += self._format_line(first, kernel)
163 return output
164
165 """
166 Returns the computed dispatch table(str). Note this only include

Callers 3

test_math_autogradcpuMethod · 0.95
test_autogradotherMethod · 0.95

Calls 5

_format_headerMethod · 0.95
rawRegistrationsMethod · 0.95
_format_lineMethod · 0.95
anyFunction · 0.50
splitMethod · 0.45

Tested by 3

test_math_autogradcpuMethod · 0.76
test_autogradotherMethod · 0.76