Searched defs:node_state (Results 1 – 6 of 6) sorted by relevance
94 const std::unordered_map<const NodeDef*, NodeState>* node_state) { in Init()150 const std::unordered_map<const NodeDef*, NodeState>* node_state) { in Init()475 auto& node_state = node_map_[node]; in MaybeUpdateInputOutput() local632 const auto& node_state = node_map_.at(node); in GetCurrNode() local662 auto& node_state = it->second; in GetNodeStateOrCreateIt() local723 auto& node_state = node_map_[node]; in AddOutputNodesToReadyQueue() local749 auto& node_state = node_map_[node]; in MarkCurrNodeExecuted() local
136 const std::unordered_map<const NodeDef*, NodeState>* node_state) {} in Init()147 void Init(const std::unordered_map<const NodeDef*, NodeState>* node_state) in Init()169 void Init(const std::unordered_map<const NodeDef*, NodeState>* node_state) in Init()
2487 const NodeState& node_state = nodedef_node_state.second; in TEST_F() local
61 absl::flat_hash_map<int, NodeState> node_state; in DfsTraversal() local
109 NodeState node_state(Handle h) { in node_state() function
983 PendingCounts::NodeState node_state(PendingCounts::Handle h) { in node_state() function