| 211 | } |
| 212 | |
| 213 | void StartCollectOp(const NodeItem* item, KernelStatsInfo* stat) { |
| 214 | if (!collect_kernel_stats || |
| 215 | collect_stats_done_ || |
| 216 | !collect_op_cost_) { |
| 217 | return; |
| 218 | } |
| 219 | |
| 220 | stat->op_start_time_ = Env::Default()->NowNanos(); |
| 221 | task_count_[item->node->id()] = 0; |
| 222 | } |
| 223 | |
| 224 | void StopCollectOp(const NodeItem* item, KernelStatsInfo* stat) { |
| 225 | if (!collect_kernel_stats || |
no test coverage detected