Home
last modified time | relevance | path

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

/external/pytorch/aten/src/ATen/
DTensorSubclassLikeUtils.h51 if (c10::impl::dispatch_mode_enabled()) in isTensorSubclassLike()
58 if (c10::impl::dispatch_mode_enabled()) in areAnyTensorSubclassLike()
65 if (c10::impl::dispatch_mode_enabled()) in areAnyOptionalTensorSubclassLike()
/external/pytorch/aten/src/ATen/core/
DTorchDispatchUtils.h15 using c10::impl::dispatch_mode_enabled;
/external/pytorch/c10/core/impl/
DTorchDispatchModeTLS.h63 C10_API bool dispatch_mode_enabled();
DTorchDispatchModeTLS.cpp180 bool dispatch_mode_enabled() { in dispatch_mode_enabled() function
/external/pytorch/torch/csrc/autograd/
Dautograd_not_implemented_fallback.cpp383 at::impl::dispatch_mode_enabled() || in autogradNotImplementedFallbackImpl()
431 if (!at::impl::dispatch_mode_enabled() && in autogradNotImplementedFallbackImpl()
443 if (!at::impl::dispatch_mode_enabled() && in autogradNotImplementedFallbackImpl()
/external/pytorch/torch/csrc/utils/
Dpython_arg_parser.cpp486 : c10::impl::dispatch_mode_enabled(); in handle_torch_function_no_python_arg_parser()