Searched defs:torch_function (Results 1 – 4 of 4) sorted by relevance
75 def implements_diagonal(torch_function): argument186 def implements_sub(torch_function): argument251 def implements_sub_diagonal(torch_function): argument315 def implements_tensor_like(torch_function): argument
81 def implements_per_sample_grads(torch_function): argument
291 py::object torch_function = in dispatch_on_subclass() local384 py::object torch_function = in dispatch_on_mode() local
1144 def temporary_tensor_subclass(torch_function=None): argument