Home
last modified time | relevance | path

Searched refs:CompositeNodeManager (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/costs/
Dvirtual_scheduler.h284 class CompositeNodeManager : public ReadyNodeManager {
286 CompositeNodeManager();
287 ~CompositeNodeManager() override {} in ~CompositeNodeManager()
Dvirtual_scheduler.cc268 CompositeNodeManager::CompositeNodeManager() in CompositeNodeManager() function in tensorflow::grappler::CompositeNodeManager
271 Status CompositeNodeManager::Init( in Init()
280 void CompositeNodeManager::AddNode(const NodeDef* node) { in AddNode()
291 const NodeDef* CompositeNodeManager::GetCurrNode() { in GetCurrNode()
344 void CompositeNodeManager::RemoveCurrNode() { in RemoveCurrNode()
358 bool CompositeNodeManager::Empty() const { in Empty()
376 return absl::make_unique<CompositeNodeManager>(); in ReadyNodeManagerFactory()
Dvirtual_scheduler_test.cc431 CompositeNodeManager manager; in TEST_F()
439 CompositeNodeManager manager; in TEST_F()
469 CompositeNodeManager manager; in TEST_F()
548 CompositeNodeManager manager; in TEST_F()
550 CompositeNodeManager manager2; in TEST_F()
2385 CompositeNodeManager composite_node_manager_;