Searched refs:ThreadLocalDistAutogradContext (Results 1 – 4 of 4) sorted by relevance
273 ThreadLocalDistAutogradContext::ThreadLocalDistAutogradContext( in ThreadLocalDistAutogradContext() function in torch::distributed::autograd::ThreadLocalDistAutogradContext279 ThreadLocalDistAutogradContext::~ThreadLocalDistAutogradContext() { in ~ThreadLocalDistAutogradContext()284 ContextPtr ThreadLocalDistAutogradContext::getContextPtr() { in getContextPtr()
159 class TORCH_API ThreadLocalDistAutogradContext {162 explicit ThreadLocalDistAutogradContext(ContextPtr&& new_context);163 ~ThreadLocalDistAutogradContext();
183 using torch::distributed::autograd::ThreadLocalDistAutogradContext; in Reducer()194 ThreadLocalDistAutogradContext::getContextPtr()); in Reducer()1068 using torch::distributed::autograd::ThreadLocalDistAutogradContext; in initialize_buckets()1069 this->rpc_context_.set(ThreadLocalDistAutogradContext::getContextPtr()); in initialize_buckets()
47 ThreadLocalDistAutogradContext contextGuard{ContextPtr(autogradContext_)}; in operator ()()