def METHOD_NAME(self):
    cuda_mem = StatefulTensor.GST_MGR.total_mem['cuda']
    cpu_mem = StatefulTensor.GST_MGR.total_mem['cpu']
    timestamp = get_timestamp_us()
    if self._tracing:
        self.events.append(StatefulTensorMemoryEvent(timestamp, DeviceType.CUDA, cuda_mem))
        self.events.append(StatefulTensorMemoryEvent(timestamp, DeviceType.CPU, cpu_mem))