Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/grappler/costs/
Dvirtual_scheduler.h267 Status SetPriority(const std::unordered_map<string, int>& node_priority);
Dvirtual_scheduler.cc263 const std::unordered_map<string, int>& node_priority) { in SetPriority() argument
264 node_priority_ = node_priority; in SetPriority()
Dvirtual_scheduler_test.cc400 std::unordered_map<string, int> node_priority = { in TEST_F() local
402 TF_EXPECT_OK(manager.SetPriority(node_priority)); in TEST_F()