Searched refs:GetCurrNode (Results 1 – 4 of 4) sorted by relevance
1619 OpContext op_context = scheduler_->GetCurrNode(); in RunScheduler()1732 EXPECT_EQ("Node1", manager.GetCurrNode()->name()); in TEST_F()1761 EXPECT_EQ("Node1", manager.GetCurrNode()->name()); in TEST_F()1763 EXPECT_EQ("Node2", manager.GetCurrNode()->name()); in TEST_F()1765 EXPECT_EQ("Node3", manager.GetCurrNode()->name()); in TEST_F()1767 EXPECT_EQ("Node4", manager.GetCurrNode()->name()); in TEST_F()1785 EXPECT_EQ("Node1", manager.GetCurrNode()->name()); in TEST_F()1787 EXPECT_EQ("Node2", manager.GetCurrNode()->name()); in TEST_F()1791 EXPECT_EQ("Node2", manager.GetCurrNode()->name()); in TEST_F()1793 EXPECT_EQ("Node3", manager.GetCurrNode()->name()); in TEST_F()[all …]
138 virtual const NodeDef* GetCurrNode() = 0;150 const NodeDef* GetCurrNode() override { in GetCurrNode() function172 const NodeDef* GetCurrNode() override;195 const NodeDef* GetCurrNode() override;233 const NodeDef* GetCurrNode() override;277 OpContext GetCurrNode() const;
68 const NodeDef* LIFOManager::GetCurrNode() { in GetCurrNode() function in tensorflow::grappler::LIFOManager81 GetCurrNode(); in RemoveCurrNode()112 const NodeDef* FirstReadyManager::GetCurrNode() { in GetCurrNode() function in tensorflow::grappler::FirstReadyManager125 GetCurrNode(); in RemoveCurrNode()168 const NodeDef* CompositeNodeManager::GetCurrNode() { in GetCurrNode() function in tensorflow::grappler::CompositeNodeManager179 const auto* op = ops_lifo.second.GetCurrNode(); in GetCurrNode()184 const auto* send = send_manager_.GetCurrNode(); in GetCurrNode()188 const auto* recv = recv_manager_.GetCurrNode(); in GetCurrNode()222 const auto* node = GetCurrNode(); in RemoveCurrNode()618 OpContext VirtualScheduler::GetCurrNode() const { in GetCurrNode() function in tensorflow::grappler::VirtualScheduler[all …]
66 for (const string& input : node_manager->GetCurrNode()->input()) { in AddCostNode()72 if (!IsMerge(*node_manager->GetCurrNode())) in AddCostNode()166 OpContext op_context = scheduler_->GetCurrNode(); in PredictCosts()