Home
last modified time | relevance | path

Searched refs:check_has_torch_function (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/autograd/
H A Dpython_variable.cpp911 if (check_has_torch_function((PyObject*)self)) { in getter()
977 if (check_has_torch_function((PyObject*)self)) { in THPVariable_get_cdata()
987 if (check_has_torch_function((PyObject*)self)) { in THPVariable_get_version()
997 if (check_has_torch_function((PyObject*)self)) { in THPVariable_get_grad_fn()
1013 if (check_has_torch_function((PyObject*)self)) { in THPVariable_set_grad_fn()
1025 if (check_has_torch_function((PyObject*)self)) { in THPVariable_is_leaf()
1034 if (check_has_torch_function((PyObject*)self)) { in THPVariable_set_data()
1051 if (check_has_torch_function((PyObject*)self)) { in THPVariable_set_grad()
1109 if (check_has_torch_function((PyObject*)self)) { in THPVariable_get_volatile()
1122 if (check_has_torch_function((PyObject*)self)) { in THPVariable_set_volatile()
[all …]
H A Dpython_variable_indexing.cpp39 if (check_has_torch_function(self)) { in THPVariable_length()
69 if (check_has_torch_function(obj)) in count_specified_dimensions()
338 if (check_has_torch_function(self)) { in THPVariable_getitem()
435 if ((check_has_torch_function(self)) || in THPVariable_setitem()
436 (check_has_torch_function(py_value))) { in THPVariable_setitem()
/aosp_15_r20/external/pytorch/tools/autograd/templates/
H A Dpython_variable_methods.cpp69 if (check_has_torch_function(self)) { in THPVariable__is_view()
86 if (check_has_torch_function(self)) { in THPVariable_apply_()
177 if (check_has_torch_function(self_)) { in THPVariable_get_device()
188 if (check_has_torch_function(self_)) { in THPVariable_has_names()
200 if (check_has_torch_function(self_)) { in THPVariable_data_ptr()
212 if (check_has_torch_function(self_)) { in THPVariable_storage_offset()
224 if (check_has_torch_function(self)) { in THPVariable_dim()
236 if (check_has_torch_function(self)) { in THPVariable_numel()
327 if (check_has_torch_function(self)) { in THPVariable_float_scalar()
338 if (check_has_torch_function(self)) { in THPVariable_complex_scalar()
[all …]
/aosp_15_r20/external/pytorch/torch/csrc/utils/
H A Ddisable_torch_function.cpp316 auto check_has_torch_function(PyObject* obj, bool ignore_mode) -> bool { in check_has_torch_function() function
331 if (torch::check_has_torch_function(obj)) { in sequence_has_torch_function()
340 if (torch::check_has_torch_function(args[i])) { in array_has_torch_function()
373 if (torch::check_has_torch_function(obj)) { in THPModule_has_torch_function_unary()
H A Ddisable_torch_function.h19 bool check_has_torch_function(PyObject* obj, bool ignore_mode = false);
H A Dpython_arg_parser.cpp759 if (check_has_torch_function(obj, /*ignore_mode*/ true)) { in is_tensor_and_append_overloaded()
1493 if (self != nullptr && check_has_torch_function(self, /*ignore_mode*/ true)) { in parse()
/aosp_15_r20/external/pytorch/tools/autograd/
H A Dgen_python_functions.py950 check_has_torch_function=gen_has_torch_function_check(