Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/utils/
H A Dpython_arg_parser.cpp286 const char* torch_function_name_str, in dispatch_on_subclass() argument
292 PyObject_FastGetAttrString(arg, torch_function_name_str); in dispatch_on_subclass()
313 torch_function_name_str, in dispatch_on_subclass()
367 const char* torch_function_name_str) { in dispatch_on_mode() argument
385 PyObject_FastGetAttrString(mode_obj.ptr(), torch_function_name_str); in dispatch_on_mode()
395 torch_function_name_str, in dispatch_on_mode()
423 torch_function_name_str, in dispatch_on_mode()
431 torch_function_name_str, in dispatch_on_mode()
453 const char* torch_function_name_str = nullptr; in handle_torch_function_no_python_arg_parser() local
456 torch_function_name_str = "__torch_function__"; in handle_torch_function_no_python_arg_parser()
[all …]