Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/utils/
H A Dpython_arg_parser.cpp452 TorchFunctionName torch_function_name) -> PyObject* { in handle_torch_function_no_python_arg_parser() argument
454 switch (torch_function_name) { in handle_torch_function_no_python_arg_parser()
462 TORCH_INTERNAL_ASSERT(0, static_cast<int>(torch_function_name)); in handle_torch_function_no_python_arg_parser()
480 torch_function_name == TorchFunctionName::TorchFunction; in handle_torch_function_no_python_arg_parser()
H A Dpython_arg_parser.h1219 TorchFunctionName torch_function_name = TorchFunctionName::TorchFunction)