Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/utils/
H A Dpython_arg_parser.cpp285 bool is_torch_function, in dispatch_on_subclass() argument
307 if (is_torch_function && in dispatch_on_subclass()
318 if (!is_torch_function) { in dispatch_on_subclass()
366 bool is_torch_function, in dispatch_on_mode() argument
375 if (is_torch_function) { in dispatch_on_mode()
398 if (!is_torch_function) { in dispatch_on_mode()
479 const bool is_torch_function = in handle_torch_function_no_python_arg_parser() local
482 return is_torch_function in handle_torch_function_no_python_arg_parser()
534 is_torch_function, in handle_torch_function_no_python_arg_parser()
553 is_torch_function, in handle_torch_function_no_python_arg_parser()
/aosp_15_r20/external/pytorch/torch/_functorch/_aot_autograd/
H A Dautograd_cache.py87 def is_torch_function(target): function
103 if not is_torch_function(node.target):