| 39 | } // namespace |
| 40 | |
| 41 | void LogMemory::RecordStep(const int64 step_id, const string& handle) { |
| 42 | MemoryLogStep step; |
| 43 | step.set_step_id(step_id); |
| 44 | step.set_handle(handle); |
| 45 | OutputToLog(step); |
| 46 | } |
| 47 | |
| 48 | void LogMemory::RecordTensorAllocation(const string& kernel_name, |
| 49 | const int64 step_id, |
nothing calls this directly
no test coverage detected