Home
last modified time | relevance | path

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

/external/pytorch/torch/csrc/autograd/
Dautograd_not_implemented_fallback.h29 TORCH_API void setAutogradFallbackMode(AutogradFallbackMode mode);
Dinit.cpp464 torch::autograd::setAutogradFallbackMode( in THPAutograd_initExtension()
469 torch::autograd::setAutogradFallbackMode( in THPAutograd_initExtension()
474 torch::autograd::setAutogradFallbackMode( in THPAutograd_initExtension()
Dautograd_not_implemented_fallback.cpp52 void setAutogradFallbackMode(AutogradFallbackMode mode) { in setAutogradFallbackMode() function