Searched defs:node_state (Results 1 – 6 of 6) sorted by relevance
94 const NodeState& node_state, in UpdateDeviceAnnotationState()597 auto& node_state = node_map_[node]; in MaybeUpdateInputOutput() local757 const auto& node_state = node_map_.at(node); in CreateOpContext() local787 auto& node_state = it->second; in GetNodeStateOrCreateIt() local843 auto& node_state = node_map_[node]; in GetOutputNodes() local869 auto& node_state = node_map_[node]; in MarkNodeExecuted() local1308 auto& node_state = node_map_.at(node); in SetNodeStateTimeScheduled() local
2658 const NodeState& node_state = nodedef_node_state.second; in TEST_F() local
61 absl::flat_hash_map<int, NodeState> node_state; in DfsTraversal() local
161 PendingCounts::NodeState node_state(PendingCounts::Handle h) { in node_state() function
121 NodeState node_state(Handle h) { in node_state() function
764 absl::flat_hash_map<int, NodeState> node_state; in DfsTypeTraversal() local