Home
last modified time | relevance | path

Searched refs:node_state_ (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/costs/
Dvirtual_scheduler.cc97 node_state_ = node_state; in Init()
101 if (node_state_->at(a).time_ready == node_state_->at(b).time_ready) { in Init()
107 return node_state_->at(a).time_ready > node_state_->at(b).time_ready; in Init()
151 node_state_ = node_state; in Init()
163 const auto& device = node_state_->at(node).device_name; in AddNode()
180 candidates.emplace_back(op, node_state_->at(op).time_ready); in GetCurrNode()
185 candidates.emplace_back(send, node_state_->at(send).time_ready); in GetCurrNode()
189 candidates.emplace_back(recv, node_state_->at(recv).time_ready); in GetCurrNode()
228 const auto device = node_state_->at(node).device_name; in RemoveCurrNode()
Dvirtual_scheduler.h216 const std::unordered_map<const NodeDef*, NodeState>* node_state_; variable
249 const std::unordered_map<const NodeDef*, NodeState>* node_state_; variable