| 64 | } |
| 65 | |
| 66 | void LogMemory::RecordTensorOutput(const string& kernel_name, |
| 67 | const int64 step_id, const int index, |
| 68 | const Tensor& tensor) { |
| 69 | MemoryLogTensorOutput output; |
| 70 | output.set_step_id(step_id); |
| 71 | output.set_kernel_name(kernel_name); |
| 72 | output.set_index(index); |
| 73 | tensor.FillDescription(output.mutable_tensor()); |
| 74 | OutputToLog(output); |
| 75 | } |
| 76 | |
| 77 | void LogMemory::RecordRawAllocation(const string& operation, |
| 78 | const int64 step_id, size_t num_bytes, |
nothing calls this directly
no test coverage detected