Package org.nd4j.linalg.api.memory

Class AllocationsTracker

java.lang.Object

org.nd4j.linalg.api.memory.AllocationsTracker

public class AllocationsTracker
extends Object

Constructor Summary

Constructors

Modifier	Constructor	Description
protected	AllocationsTracker()

Method Summary

Modifier and Type	Method	Description
long	bytesOnDevice(Integer deviceId)
long	bytesOnDevice(AllocationKind kind, Integer deviceId)
static AllocationsTracker	getInstance()
void	markAllocated(AllocationKind kind, Integer deviceId, long bytes)
void	markReleased(AllocationKind kind, Integer deviceId, long bytes)
protected DeviceAllocationsTracker	trackerForDevice(Integer deviceId)

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Detail

AllocationsTracker

protected AllocationsTracker()

Method Detail

getInstance

public static AllocationsTracker getInstance()

trackerForDevice

protected DeviceAllocationsTracker trackerForDevice(Integer deviceId)

markAllocated

public void markAllocated(AllocationKind kind,
                          Integer deviceId,
                          long bytes)

markReleased

public void markReleased(AllocationKind kind,
                         Integer deviceId,
                         long bytes)

bytesOnDevice

public long bytesOnDevice(Integer deviceId)

bytesOnDevice

public long bytesOnDevice(AllocationKind kind,
                          Integer deviceId)