| 144 | } |
| 145 | |
| 146 | void NodeExecStatsWrapper::SetScheduled(int64 nanos) { |
| 147 | stats_->set_scheduled_micros(nanos / EnvTime::kMicrosToNanos); |
| 148 | stats_->set_scheduled_nanos(nanos); |
| 149 | } |
| 150 | |
| 151 | void NodeExecStatsWrapper::SetMemory(OpKernelContext* ctx) { |
| 152 | for (const auto& allocator_pair : ctx->ConsumeWrappedAllocators()) { |