Searched refs:completion_times (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | static_schedule_test.cc | 56 std::unordered_map<const NodeDef*, Costs::NanoSeconds> completion_times; in TEST_F() local
58 EstimateEarliestExecutionTimes(item, cluster.get(), &completion_times); in TEST_F()
61 EXPECT_EQ(item.graph.node_size(), completion_times.size()); in TEST_F()
63 for (auto time : completion_times) { in TEST_F()
102 std::unordered_map<const NodeDef*, Costs::NanoSeconds> completion_times; in TEST_F() local
104 EstimateEarliestExecutionTimes(item, cluster.get(), &completion_times); in TEST_F()
107 EXPECT_EQ(item.graph.node_size(), completion_times.size()); in TEST_F()
109 for (auto time : completion_times) { in TEST_F()
|
| D | static_schedule.cc | 61 std::unordered_map<const NodeDef*, Costs::NanoSeconds>* completion_times) { in EstimateEarliestExecutionTimes() argument
69 (*completion_times)[&node] = 0; in EstimateEarliestExecutionTimes()
106 execution_time + (*completion_times)[node]; in EstimateEarliestExecutionTimes()
107 (*completion_times)[node] = completion_time; in EstimateEarliestExecutionTimes()
120 std::max(completion_time, (*completion_times)[fanout]); in EstimateEarliestExecutionTimes()
121 (*completion_times)[fanout] = ready_time; in EstimateEarliestExecutionTimes()
|