Searched refs:execution_times (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | static_schedule.h | 36 std::unordered_map<const NodeDef*, Costs::NanoSeconds>* execution_times);
44 execution_times,
|
| D | static_schedule_test.cc | 132 std::unordered_map<const NodeDef*, Costs::NanoSeconds> execution_times; in TEST_F() local
134 execution_times[&node] = 0; in TEST_F()
137 Status status = EstimateRequiredTimes(item, cluster.get(), execution_times, in TEST_F()
|
| D | static_schedule.cc | 131 execution_times, in EstimateRequiredTimes() argument
155 auto it = execution_times.find(&node); in EstimateRequiredTimes()
156 if (it != execution_times.end()) { in EstimateRequiredTimes()
|
| D | memory_optimizer.cc | 778 execution_times) { in FindSwapInTrigger() argument
792 auto it2 = execution_times.find(input_node); in FindSwapInTrigger()
793 if (it2 == execution_times.end()) { in FindSwapInTrigger()
829 auto it2 = execution_times.find(input_node); in FindSwapInTrigger()
830 if (it2 == execution_times.end()) { in FindSwapInTrigger()
895 execution_times) { in FindSwapOutTrigger() argument
914 auto it = execution_times.find(port.node); in FindSwapOutTrigger()
915 if (it != execution_times.end() && it->second < earliest_fanout) { in FindSwapOutTrigger()
1163 std::unordered_map<const NodeDef*, Costs::NanoSeconds> execution_times; in SwappingPass() local
1164 if (!EstimateEarliestExecutionTimes(*item, cluster, &execution_times).ok()) { in SwappingPass()
[all …]
|