Package 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	Description
DeviceAllocationsTracker()

Method Summary

Modifier and Type	Method	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)