Searched refs:TorchFunctionName (Results 1 – 3 of 3) sorted by relevance
245 TorchFunctionName::TorchFunction); in handle_torch_function()452 TorchFunctionName torch_function_name) -> PyObject* { in handle_torch_function_no_python_arg_parser()455 case TorchFunctionName::TorchFunction: in handle_torch_function_no_python_arg_parser()458 case TorchFunctionName::TorchDispatch: in handle_torch_function_no_python_arg_parser()480 torch_function_name == TorchFunctionName::TorchFunction; in handle_torch_function_no_python_arg_parser()
1210 enum class TorchFunctionName { TorchFunction, TorchDispatch }; enum1219 TorchFunctionName torch_function_name = TorchFunctionName::TorchFunction)
227 TorchFunctionName::TorchDispatch));351 TorchFunctionName::TorchDispatch); in dispatch()