Searched refs:node_state (Results 1 – 8 of 8) sorted by relevance
94 const NodeState& node_state, in UpdateDeviceAnnotationState() argument108 node_state.shape_incompatible ? 1 : 0; in UpdateDeviceAnnotationState()597 auto& node_state = node_map_[node]; in MaybeUpdateInputOutput() local598 auto& inputs = node_state.input_properties; in MaybeUpdateInputOutput()599 auto& outputs = node_state.output_properties; in MaybeUpdateInputOutput()757 const auto& node_state = node_map_.at(node); in CreateOpContext() local759 op_context.device_name = node_state.device_name; in CreateOpContext()763 for (auto& input : node_state.input_properties) { in CreateOpContext()766 for (auto& output : node_state.output_properties) { in CreateOpContext()787 auto& node_state = it->second; in GetNodeStateOrCreateIt() local[all …]
2658 const NodeState& node_state = nodedef_node_state.second; in TEST_F() local2660 bn_node = &node_state; in TEST_F()2663 x_node = &node_state; in TEST_F()
132 EXPECT_EQ(c.node_state(h), PendingCounts::PENDING_NOTREADY); in TEST()135 EXPECT_EQ(c.node_state(h), PendingCounts::PENDING_READY); in TEST()137 EXPECT_EQ(c.node_state(h), PendingCounts::STARTED); in TEST()139 EXPECT_EQ(c.node_state(h), PendingCounts::COMPLETED); in TEST()
161 PendingCounts::NodeState node_state(PendingCounts::Handle h) { in node_state() function162 return counts.node_state(h); in node_state()
188 if (iteration->node_state(pending_id) == PendingCounts::PENDING_NOTREADY || in DumpIterationState()189 iteration->node_state(pending_id) == PendingCounts::PENDING_READY) { in DumpIterationState()197 if (iteration->node_state(pending_id) == PendingCounts::STARTED) { in DumpIterationState()
121 NodeState node_state(Handle h) { in node_state() function
61 absl::flat_hash_map<int, NodeState> node_state; in DfsTraversal() local66 NodeState& state = node_state[w.node]; in DfsTraversal()
764 absl::flat_hash_map<int, NodeState> node_state; in DfsTypeTraversal() local769 NodeState& state = node_state[w.node]; in DfsTypeTraversal()