org.nd4j.linalg.util

Class DeviceLocal<T>

java.lang.Object

org.nd4j.linalg.util.DeviceLocal<T>

Direct Known Subclasses:

DeviceLocalNDArray

public class DeviceLocal<T>
extends Object

Class similar to Java ThreadLocal class, but locality is preserved with respect to device used

Author:

raver119@gmail.com

Constructor Summary

Constructors

Constructor and Description
DeviceLocal()

Method Summary

Modifier and Type	Method and Description
void	clear() This method removes object stored for current device
T	get() This method returns object local to current deviceId
T	get(int deviceId) This method returns object local to target device
void	set(int deviceId, T object) This method sets object for specific device
void	set(T object) This method sets object for current device

Methods inherited from class java.lang.Object

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

Constructor Detail

DeviceLocal

public DeviceLocal()

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