Searched refs:torch_function_mode_enabled (Results 1 – 8 of 8) sorted by relevance
32 TORCH_API bool torch_function_mode_enabled();
44 bool torch_function_mode_enabled() { in torch_function_mode_enabled() function
875 if (at::impl::torch_function_mode_enabled()) { in set_grad_enabled()929 if (at::impl::torch_function_mode_enabled()) { in set_multithreading_enabled()967 if (at::impl::torch_function_mode_enabled()) { in set_view_replay_enabled()1071 if (at::impl::torch_function_mode_enabled()) { in is_torch_function_mode_enabled()
317 if (!ignore_mode && at::impl::torch_function_mode_enabled()) in check_has_torch_function()
382 return !overloaded_args.empty() || at::impl::torch_function_mode_enabled(); in has_torch_function()
483 ? at::impl::torch_function_mode_enabled() in handle_torch_function_no_python_arg_parser()
851 if (!overloaded_args.empty() || at::impl::torch_function_mode_enabled()) { in _maybe_handle_torch_function()
369 self.torch_function_mode_enabled = torch._C._is_torch_function_mode_enabled()