Searched refs:start_times (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | scheduler.cc | 212 std::vector<Microseconds>* start_times) { in ComputeSchedule() argument
248 if ((*start_times)[out->id()] < event.time + copy_time) { in ComputeSchedule()
249 (*start_times)[out->id()] = event.time + copy_time; in ComputeSchedule()
254 Event e{out, (*start_times)[out->id()], false}; in ComputeSchedule()
272 (*start_times)[e.node->id()] = event.time; in ComputeSchedule()
299 std::vector<Microseconds>* start_times) { in ComputeSchedule() argument
304 return greedysched.ComputeSchedule(start_times); in ComputeSchedule()
309 std::vector<Microseconds> start_times; in AssignPriorities() local
310 Microseconds makespan = ComputeSchedule(&start_times); in AssignPriorities()
313 (*priorities)[n->id()] = start_times[n->id()].value(); in AssignPriorities()
|
| D | scheduler.h | 78 Microseconds ComputeSchedule(std::vector<Microseconds>* start_times);
102 Microseconds ComputeSchedule(std::vector<Microseconds>* start_times);
|
| D | master_session.cc | 429 sa.ComputeAsap(&popts.start_times); in DoBuildPartitions()
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | graph_partition.cc | 833 std::vector<std::pair<const NodeDef*, int64>> start_times; in TopologicalSortNodesWithTimePriority() local
834 start_times.reserve(gdef->node_size()); in TopologicalSortNodesWithTimePriority()
839 start_times.emplace_back(cur.node, cur.start_time); in TopologicalSortNodesWithTimePriority()
853 nodes->swap(start_times); in TopologicalSortNodesWithTimePriority()
867 std::vector<std::pair<const NodeDef*, int64>> start_times; in AddControlEdges() local
869 status = TopologicalSortNodesWithTimePriority(gdef, &start_times, in AddControlEdges()
878 int64 makespan = start_times.back().second; in AddControlEdges()
884 while (i < num_epochs && static_cast<size_t>(j) < start_times.size()) { in AddControlEdges()
885 if (i * resolution > start_times[j].second) { in AddControlEdges()
895 string src_name = start_times[j - 1].first->name(); in AddControlEdges()
[all …]
|
| D | graph_partition.h | 78 std::vector<Microseconds> start_times; member
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | virtual_scheduler_test.cc | 1697 const std::unordered_map<string, int64>& start_times, in ValidateDependencyChain() argument
1701 int64 curr_node_time = start_times.at(node); in ValidateDependencyChain()
2560 std::unordered_map<string, int64> start_times; in TEST_F() local
2563 start_times[stats.node_name()] = stats.all_start_micros(); in TEST_F()
2601 start_times, in TEST_F()
2606 ValidateDependencyChain(start_times, in TEST_F()
2610 ValidateDependencyChain(start_times, {"while/Switch", "while/Exit"}); in TEST_F()
2612 start_times, {"while/Identity", "while/concat/axis", "while/concat"}); in TEST_F()
2613 ValidateDependencyChain(start_times, {"while/Identity", "while/add"}); in TEST_F()
2614 ValidateDependencyChain(start_times, {"while/Switch_1", "while/Exit_1"}); in TEST_F()
|