Searched refs:ready_nodes_ (Results 1 – 2 of 2) sorted by relevance
1323 ready_nodes_(ready_nodes) {} in VirtualScheduler()1328 : scheduler_state_(std::move(scheduler_state)), ready_nodes_(ready_nodes) {} in VirtualScheduler()1335 TF_RETURN_IF_ERROR(ready_nodes_->Init(GetNodeStates())); in Init()1341 ready_nodes_->AddNode(node); in Init()1348 const NodeDef* node = ready_nodes_->GetCurrNode(); in GetCurrNode()1354 const NodeDef* node = ready_nodes_->GetCurrNode(); in MarkCurrNodeExecuted()1357 scheduler_state_->CreateOpContext(ready_nodes_->GetCurrNode())); in MarkCurrNodeExecuted()1360 ready_nodes_->AddNode(node); in MarkCurrNodeExecuted()1362 ready_nodes_->RemoveCurrNode(); in MarkCurrNodeExecuted()1363 return !ready_nodes_->Empty(); in MarkCurrNodeExecuted()
521 ReadyNodeManager* ready_nodes_; // Not owned. variable