Home
last modified time | relevance | path

Searched refs:NanoSeconds (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/costs/
Dcost_estimator_test.cc26 c.execution_time = Costs::NanoSeconds(1); in TEST()
27 c.compute_time = Costs::NanoSeconds(2); in TEST()
28 c.memory_time = Costs::NanoSeconds(3); in TEST()
29 c.intermediate_memory_time = Costs::NanoSeconds(4); in TEST()
30 c.intermediate_memory_read_time = Costs::NanoSeconds(5); in TEST()
31 c.intermediate_memory_write_time = Costs::NanoSeconds(6); in TEST()
41 EXPECT_EQ(sum.execution_time, Costs::NanoSeconds(2)); in TEST()
42 EXPECT_EQ(sum.compute_time, Costs::NanoSeconds(4)); in TEST()
43 EXPECT_EQ(sum.memory_time, Costs::NanoSeconds(6)); in TEST()
44 EXPECT_EQ(sum.intermediate_memory_time, Costs::NanoSeconds(8)); in TEST()
[all …]
Dcost_estimator.h100 struct NanoSeconds : std::chrono::nanoseconds { struct
101 NanoSeconds() : std::chrono::nanoseconds(0) {} in NanoSeconds() argument
102 NanoSeconds(double d) : std::chrono::nanoseconds(static_cast<int64>(d)) {} in NanoSeconds() function
103 NanoSeconds(const std::chrono::nanoseconds& d) in NanoSeconds() function
105 NanoSeconds& operator=(const std::chrono::nanoseconds& d) {
115 static NanoSeconds infinity() { in infinity() argument
116 return NanoSeconds(std::chrono::nanoseconds::max()); in infinity()
122 typedef NanoSeconds Duration;
174 inline std::ostream& operator<<(std::ostream& os, const Costs::NanoSeconds d) {
Dgraph_memory.cc197 Costs::NanoSeconds(1) + in InferFromTrace()
238 Costs::NanoSeconds(1) + in InferFromTrace()
Danalytical_cost_estimator_test.cc105 EXPECT_EQ(Costs::NanoSeconds(9157), summary.execution_time); in TEST_F()
Dop_level_cost_estimator.cc523 Costs::NanoSeconds compute_cost(std::ceil(operations / device_info.gigaops)); in PredictOpCountBasedCost()
527 Costs::NanoSeconds memory_cost( in PredictOpCountBasedCost()
545 Costs::NanoSeconds intermediate_memory_cost = in PredictOpCountBasedCost()
558 Costs::NanoSeconds(intermediate_read_time); in PredictOpCountBasedCost()
560 Costs::NanoSeconds(intermediate_write_time); in PredictOpCountBasedCost()
Dvirtual_scheduler.cc922 const Costs::NanoSeconds wall_time_ns = state.GetCurrTime(); in Summary()
951 Costs::NanoSeconds total_compute_time_ns; in Summary()
Dvirtual_scheduler_test.cc1608 c.execution_time = Costs::NanoSeconds(exec_cost); in SimplePredictCosts()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dstatic_schedule_test.cc56 std::unordered_map<const NodeDef*, Costs::NanoSeconds> completion_times; in TEST_F()
65 EXPECT_EQ(Costs::NanoSeconds(1), time.second); in TEST_F()
67 EXPECT_EQ(Costs::NanoSeconds(1500001), time.second); in TEST_F()
69 EXPECT_EQ(Costs::NanoSeconds(4000004), time.second); in TEST_F()
71 EXPECT_EQ(Costs::NanoSeconds(6500007), time.second); in TEST_F()
73 EXPECT_EQ(Costs::NanoSeconds(9000010), time.second); in TEST_F()
75 EXPECT_EQ(Costs::NanoSeconds(11500013), time.second); in TEST_F()
77 EXPECT_EQ(Costs::NanoSeconds(14000013), time.second); in TEST_F()
102 std::unordered_map<const NodeDef*, Costs::NanoSeconds> completion_times; in TEST_F()
111 EXPECT_EQ(Costs::NanoSeconds(1), time.second); in TEST_F()
[all …]
Dstatic_schedule.cc30 static Costs::NanoSeconds PredictExecutionTime( in PredictExecutionTime()
52 Costs::NanoSeconds estimate = in PredictExecutionTime()
56 return std::max(estimate, Costs::NanoSeconds(1)); in PredictExecutionTime()
61 std::unordered_map<const NodeDef*, Costs::NanoSeconds>* completion_times) { in EstimateEarliestExecutionTimes()
103 Costs::NanoSeconds execution_time = in EstimateEarliestExecutionTimes()
105 Costs::NanoSeconds completion_time = in EstimateEarliestExecutionTimes()
119 Costs::NanoSeconds ready_time = in EstimateEarliestExecutionTimes()
130 const std::unordered_map<const NodeDef*, Costs::NanoSeconds>& in EstimateRequiredTimes()
132 std::unordered_map<const NodeDef*, Costs::NanoSeconds>* required_times) { in EstimateRequiredTimes()
136 (*required_times)[&node] = Costs::NanoSeconds::max(); in EstimateRequiredTimes()
[all …]
Dstatic_schedule.h36 std::unordered_map<const NodeDef*, Costs::NanoSeconds>* execution_times);
43 const std::unordered_map<const NodeDef*, Costs::NanoSeconds>&
45 std::unordered_map<const NodeDef*, Costs::NanoSeconds>* required_times);
Dmemory_optimizer.cc771 Costs::NanoSeconds time_to_swap = 0;
777 const std::unordered_map<const NodeDef*, Costs::NanoSeconds>& in FindSwapInTrigger()
782 Costs::NanoSeconds max_trigger_time(0); in FindSwapInTrigger()
810 std::map<Costs::NanoSeconds, const NodeDef*> candidates; in FindSwapInTrigger()
894 const std::unordered_map<const NodeDef*, Costs::NanoSeconds>& in FindSwapOutTrigger()
908 Costs::NanoSeconds earliest_fanout(Costs::NanoSeconds::infinity()); in FindSwapOutTrigger()
979 std::unordered_map<string, Costs::NanoSeconds> op_completion_times; in IdentifySwappingCandidates()
996 Costs::NanoSeconds exec_time = in IdentifySwappingCandidates()
997 Costs::NanoSeconds(1) + in IdentifySwappingCandidates()
1163 std::unordered_map<const NodeDef*, Costs::NanoSeconds> execution_times; in SwappingPass()