Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/grappler/costs/
Dvirtual_scheduler.h368 const std::unordered_map<const NodeDef*, NodeState>* GetNodeStates() const { in GetNodeStates() function
509 const std::unordered_map<const NodeDef*, NodeState>* GetNodeStates() const { in GetNodeStates() function
510 return scheduler_state_->GetNodeStates(); in GetNodeStates()
Dvirtual_scheduler.cc1335 TF_RETURN_IF_ERROR(ready_nodes_->Init(GetNodeStates())); in Init()
Dvirtual_scheduler_test.cc2653 const auto* node_states = scheduler_->GetNodeStates(); in TEST_F()