Package org.nd4j.linalg.util
Class DeviceLocal<T>
- java.lang.Object
-
- org.nd4j.linalg.util.DeviceLocal<T>
-
- Direct Known Subclasses:
DeviceLocalNDArray
public abstract class DeviceLocal<T> extends Object
-
-
Field Summary
Fields Modifier and Type Field Description protected Map<Integer,T>
backingMap
protected INDArray
delayedArray
protected boolean
delayedMode
protected int
lastSettledDevice
protected List<ReentrantReadWriteLock>
locksMap
protected List<AtomicInteger>
updatesMap
-
Constructor Summary
Constructors Constructor Description DeviceLocal(boolean delayedMode)
-
Method Summary
All Methods Instance Methods Concrete Methods Modifier and Type Method Description void
clear()
This method removes object stored for current deviceT
get()
This method returns object local to current deviceIdT
get(int deviceId)
This method returns object local to target devicevoid
set(int deviceId, T object)
This method sets object for specific devicevoid
set(T object)
This method sets object for current device
-
-
-
Field Detail
-
locksMap
protected List<ReentrantReadWriteLock> locksMap
-
updatesMap
protected List<AtomicInteger> updatesMap
-
delayedMode
protected final boolean delayedMode
-
delayedArray
protected volatile INDArray delayedArray
-
lastSettledDevice
protected int lastSettledDevice
-
-
Method Detail
-
get
@Nullable public T get()
This method returns object local to current deviceId- Returns:
-
get
@Nullable public T get(int deviceId)
This method returns object local to target device- Parameters:
deviceId
-- Returns:
-
set
public void set(int deviceId, T object)
This method sets object for specific device- Parameters:
deviceId
-object
-
-
set
public void set(T object)
This method sets object for current device- Parameters:
object
-
-
clear
public void clear()
This method removes object stored for current device
-
-