org.deeplearning4j.optimize.solvers.accumulation

Class IndexedTail

java.lang.Object

org.deeplearning4j.optimize.solvers.accumulation.IndexedTail

public class IndexedTail
extends Object

This class provides queue-like functionality for multiple readers/multiple writers, with transparent duplication and collapsing ability

Author:

[email protected]

Field Summary

Fields

Modifier and Type	Field and Description
protected boolean	allowCollapse
protected AtomicLong	collapsedIndex
protected AtomicBoolean	collapsedMode
protected int	collapseThreshold
protected AtomicBoolean	dead
protected int	expectedConsumers
protected AtomicLong	lastDeletedIndex
protected ReentrantReadWriteLock	lock
protected ConcurrentHashMap<Long,AtomicLong>	positions
protected long[]	shape
protected Map<Long,INDArray>	updates
protected AtomicLong	updatesCounter

Constructor Summary

Constructors

Constructor and Description
IndexedTail(int expectedConsumers)
IndexedTail(int expectedConsumers, boolean allowCollapse, long[] shape)

Method Summary

Modifier and Type	Method and Description
boolean	drainTo(INDArray array)
boolean	drainTo(long threadId, INDArray array)
protected long	firstNotAppliedIndexEverywhere()
protected long	getDelta()
protected long	getDelta(long threadId)
protected long	getGlobalPosition()
protected long	getLocalPosition()
protected long	getLocalPosition(long threadId)
boolean	hasAnything()
boolean	hasAnything(long threadId)
protected boolean	isDead()
protected void	maintenance() This method does maintenance of updates within
protected long	maxAppliedIndexEverywhere()
protected void	notifyDead()
void	purge()
void	put(INDArray update) This mehtod adds update, with optional collapse
protected INDArray	smartDecompress(INDArray encoded, INDArray target)
protected int	updatesSize() This method returns actual number of updates stored within tail

Methods inherited from class java.lang.Object

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

Field Detail

positions

protected ConcurrentHashMap<Long,AtomicLong> positions

updates

protected Map<Long,INDArray> updates

updatesCounter

protected AtomicLong updatesCounter

lastDeletedIndex

protected AtomicLong lastDeletedIndex

expectedConsumers

protected final int expectedConsumers

dead

protected AtomicBoolean dead

lock

protected ReentrantReadWriteLock lock

allowCollapse

protected final boolean allowCollapse

shape

protected final long[] shape

collapseThreshold

protected final int collapseThreshold

See Also:

Constant Field Values

collapsedMode

protected AtomicBoolean collapsedMode

collapsedIndex

protected AtomicLong collapsedIndex

Constructor Detail

IndexedTail

public IndexedTail(int expectedConsumers)

IndexedTail

public IndexedTail(int expectedConsumers,
                   boolean allowCollapse,
                   long[] shape)

Method Detail

put

public void put(@NonNull
                INDArray update)

This mehtod adds update, with optional collapse

Parameters:

update -

firstNotAppliedIndexEverywhere

protected long firstNotAppliedIndexEverywhere()

maxAppliedIndexEverywhere

protected long maxAppliedIndexEverywhere()

hasAnything

public boolean hasAnything()

hasAnything

public boolean hasAnything(long threadId)

Returns:

drainTo

public boolean drainTo(@NonNull
                       INDArray array)

getGlobalPosition

protected long getGlobalPosition()

getLocalPosition

protected long getLocalPosition()

getDelta

protected long getDelta()

getDelta

protected long getDelta(long threadId)

getLocalPosition

protected long getLocalPosition(long threadId)

drainTo

public boolean drainTo(long threadId,
                       @NonNull
                       INDArray array)

maintenance

protected void maintenance()

This method does maintenance of updates within

updatesSize

protected int updatesSize()

This method returns actual number of updates stored within tail

Returns:

smartDecompress

protected INDArray smartDecompress(INDArray encoded,
                                   @NonNull
                                   INDArray target)

isDead

protected boolean isDead()

notifyDead

protected void notifyDead()

purge

public void purge()