| 56 | } |
| 57 | |
| 58 | void LogMemory::RecordTensorDeallocation(const int64 allocation_id, |
| 59 | const string& allocator_name) { |
| 60 | MemoryLogTensorDeallocation deallocation; |
| 61 | deallocation.set_allocation_id(allocation_id); |
| 62 | deallocation.set_allocator_name(allocator_name); |
| 63 | OutputToLog(deallocation); |
| 64 | } |
| 65 | |
| 66 | void LogMemory::RecordTensorOutput(const string& kernel_name, |
| 67 | const int64 step_id, const int index, |
nothing calls this directly
no test coverage detected