Searched refs:torch_api (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/csrc/utils/ |
| H A D | python_arg_parser.cpp | 160 py::object torch_api = PyObject_FastGetAttrString( in handle_torch_function_getter() local
168 torch_api.ptr(), in handle_torch_function_getter()
176 py::object torch_api = PyObject_FastGetAttrString( in handle_torch_function_setter() local
186 torch_api.ptr(), in handle_torch_function_setter()
194 torch_api.ptr(), in handle_torch_function_setter()
231 PyObject* torch_api, in handle_torch_function() argument
234 PyObject_FastGetAttrString(torch_api, (char*)func_name.c_str()); in handle_torch_function()
598 PyObject* torch_api, in handle_torch_function() argument
602 torch_api, in handle_torch_function()
621 PyObject* torch_api, in handle_torch_function() argument
[all …]
|
| H A D | python_arg_parser.h | 1186 PyObject* torch_api,
1195 PyObject* torch_api,
1205 PyObject* torch_api = THPVariableClass,
|