org.nd4j.linalg.api.memory

Class DeviceAllocationsTracker

java.lang.Object

org.nd4j.linalg.api.memory.DeviceAllocationsTracker

public class DeviceAllocationsTracker
extends Object

Constructor Summary

Constructors

Constructor and Description
DeviceAllocationsTracker()

Method Summary

Modifier and Type	Method and Description
long	getState(@NonNull AllocationKind kind)
void	updateState(@NonNull 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
                        @NonNull AllocationKind kind,
                        long bytes)

getState

public long getState(@NonNull
                     @NonNull AllocationKind kind)