org.nd4j.linalg.api.memory

Class DeviceAllocationsTracker

java.lang.Object

org.nd4j.linalg.api.memory.DeviceAllocationsTracker

public class DeviceAllocationsTracker
extends java.lang.Object

Author:

raver119@gmail.com

Constructor Summary

Constructors

Constructor and Description
DeviceAllocationsTracker()

Method Summary

Modifier and Type	Method and Description
long	getState(AllocationKind kind)
void	updateState(AllocationKind kind, long bytes)

Methods inherited from class java.lang.Object

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

Constructor Detail

DeviceAllocationsTracker

public DeviceAllocationsTracker()

Method Detail

updateState

public void updateState(@NonNull
                        AllocationKind kind,
                        long bytes)

getState

public long getState(@NonNull
                     AllocationKind kind)