Home
last modified time | relevance | path

Searched refs:GetCurrNode (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/costs/
Dvirtual_scheduler_test.cc73 EXPECT_EQ(manager.GetCurrNode()->name(), "Node1"); in TEST_F()
96 EXPECT_EQ(manager.GetCurrNode()->name(), "Node1"); in TEST_F()
98 EXPECT_EQ(manager.GetCurrNode()->name(), "Node2"); in TEST_F()
100 EXPECT_EQ(manager.GetCurrNode()->name(), "Node3"); in TEST_F()
102 EXPECT_EQ(manager.GetCurrNode()->name(), "Node4"); in TEST_F()
117 EXPECT_EQ(manager.GetCurrNode()->name(), "Node1"); in TEST_F()
119 EXPECT_EQ(manager.GetCurrNode()->name(), "Node2"); in TEST_F()
123 EXPECT_EQ(manager.GetCurrNode()->name(), "Node2"); in TEST_F()
125 EXPECT_EQ(manager.GetCurrNode()->name(), "Node3"); in TEST_F()
127 EXPECT_EQ(manager.GetCurrNode()->name(), "Node4"); in TEST_F()
[all …]
Dvirtual_scheduler.h170 virtual const NodeDef* GetCurrNode() = 0;
180 const NodeDef* GetCurrNode() override { in GetCurrNode() function
200 const NodeDef* GetCurrNode() override;
223 const NodeDef* GetCurrNode() override;
292 const NodeDef* GetCurrNode() override;
479 virtual OpContext GetCurrNode();
Danalytical_cost_estimator.cc68 for (const string& input : node_manager->GetCurrNode()->input()) { in AddCostNode()
74 if (!IsMerge(*node_manager->GetCurrNode())) in AddCostNode()
194 OpContext op_context = scheduler_->GetCurrNode(); in PredictCosts()
Dvirtual_scheduler.cc140 const NodeDef* LIFOManager::GetCurrNode() { in GetCurrNode() function in tensorflow::grappler::LIFOManager
153 GetCurrNode(); in RemoveCurrNode()
186 const NodeDef* HeapReadyManager::GetCurrNode() { in GetCurrNode() function in tensorflow::grappler::HeapReadyManager
291 const NodeDef* CompositeNodeManager::GetCurrNode() { in GetCurrNode() function in tensorflow::grappler::CompositeNodeManager
302 const auto* op = ops_lifo.second.GetCurrNode(); in GetCurrNode()
307 const auto* send = send_manager_.GetCurrNode(); in GetCurrNode()
311 const auto* recv = recv_manager_.GetCurrNode(); in GetCurrNode()
345 const auto* node = GetCurrNode(); in RemoveCurrNode()
1347 OpContext VirtualScheduler::GetCurrNode() { in GetCurrNode() function in tensorflow::grappler::VirtualScheduler
1348 const NodeDef* node = ready_nodes_->GetCurrNode(); in GetCurrNode()
[all …]