Home
last modified time | relevance | path

Searched refs:step_stat (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/profiler/internal/
Dtfprof_node.cc46 void ExecStep::AddTimeStats(const string& dev, const NodeExecStats& step_stat) { in AddTimeStats() argument
48 if (step_stat.all_start_micros() > 0) { in AddTimeStats()
52 static_cast<int64_t>(step_stat.all_start_micros()))); in AddTimeStats()
54 exec_.set_all_start_micros(step_stat.all_start_micros()); in AddTimeStats()
56 int64_t op_end_rel_micros = step_stat.op_end_rel_micros(); in AddTimeStats()
63 step_stat.all_start_micros() + op_end_rel_micros)); in AddTimeStats()
66 std::make_pair(step_stat.all_start_micros(), op_end_rel_micros); in AddTimeStats()
81 const NodeExecStats& step_stat) { in AddMemoryStats() argument
83 if (step_stat.all_start_micros() > 0) { in AddMemoryStats()
84 exec_mem.set_memory_micros(step_stat.all_start_micros() + in AddMemoryStats()
[all …]
Dtfprof_node.h90 void AddTimeStats(const string& dev, const NodeExecStats& step_stat);
92 void AddMemoryStats(const string& dev, const NodeExecStats& step_stat);
353 const NodeExecStats& step_stat);