Searched refs:call_torch_function (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/_dynamo/variables/ |
| H A D | torch_function.py | 204 def call_torch_function( function
249 res = arg.call_torch_function(
336 return self.call_torch_function(
346 def call_torch_function(self, tx: "InstructionTranslator", fn, types, args, kwargs): member in TensorWithTFOverrideVariable
347 return call_torch_function(
|
| H A D | user_defined.py | 661 def call_torch_function(self, tx: "InstructionTranslator", fn, types, args, kwargs): member in UserDefinedObjectVariable
664 from .torch_function import _get_subclass_type_var, call_torch_function
666 return call_torch_function(
|
| /aosp_15_r20/external/pytorch/functorch/csrc/dim/ |
| H A D | dim.cpp | 3071 PyObject* call_torch_function(PyObject *self, in call_torch_function() function
3095 …nfo = WrappedOperator::create(mpy::object::borrow(orig), (PyCFunction)(void*) call_torch_function); in _wrap_method()
|