Home
last modified time | relevance | path

Searched refs:getAutogradFallbackMode (Results 1 – 3 of 3) sorted by relevance

/external/pytorch/torch/csrc/autograd/
Dautograd_not_implemented_fallback.h30 TORCH_API AutogradFallbackMode getAutogradFallbackMode();
Dautograd_not_implemented_fallback.cpp57 AutogradFallbackMode getAutogradFallbackMode() { in getAutogradFallbackMode() function
108 if (getAutogradFallbackMode() == AutogradFallbackMode::Nothing) { in basicAutogradNotImplementedFallbackImpl()
113 getAutogradFallbackMode() == AutogradFallbackMode::Warn); in basicAutogradNotImplementedFallbackImpl()
Dinit.cpp481 auto mode = torch::autograd::getAutogradFallbackMode(); in THPAutograd_initExtension()