Searched refs:graphTask_ (Results 1 – 2 of 2) sorted by relevance
117 TORCH_INTERNAL_ASSERT(graphTask_); in retrieveGraphTask()118 return graphTask_; in retrieveGraphTask()125 !graphTask_, in setGraphTask()127 graphTask_ = std::move(graphTask); in setGraphTask()132 graphTask_ = nullptr; in resetGraphTask()141 if (graphTask_) { in addOutstandingRpc()142 graphTask_->set_exception_without_signal(nullptr); in addOutstandingRpc()144 if (!graphTask_->future_completed_.exchange(true)) { in addOutstandingRpc()145 graphTask_->future_result_->setErrorIfNeeded(future.exception_ptr()); in addOutstandingRpc()
143 std::shared_ptr<torch::autograd::GraphTask> graphTask_; variable