org.deeplearning4j.parallelism

Class InplaceParallelInference.ModelHolder

java.lang.Object

org.deeplearning4j.parallelism.InplaceParallelInference.ModelHolder

Enclosing class:

InplaceParallelInference

protected static class InplaceParallelInference.ModelHolder
extends Object

Field Summary

Fields

Modifier and Type	Field and Description
protected boolean	isCG
protected boolean	isMLN
protected LoadBalanceMode	loadBalanceMode
protected ReentrantReadWriteLock	modelLock
protected AtomicLong	position
protected BlockingQueue<Model>	queue
protected List<Model>	replicas
protected boolean	rootDevice
protected Model	sourceModel
protected int	targetDeviceId
protected int	workers

Constructor Summary

Constructors

Modifier	Constructor and Description
protected	ModelHolder()

Method Summary

Modifier and Type	Method and Description
protected Model	acquireModel()
protected void	init()
protected INDArray[]	output(INDArray[] input, INDArray[] inputMasks)
protected void	releaseModel(Model model)
protected void	updateModel(@NonNull Model model)

Methods inherited from class java.lang.Object

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

Field Detail

sourceModel

protected Model sourceModel

workers

protected int workers

replicas

protected List<Model> replicas

rootDevice

protected boolean rootDevice

loadBalanceMode

protected LoadBalanceMode loadBalanceMode

targetDeviceId

protected int targetDeviceId

position

protected final AtomicLong position

modelLock

protected final ReentrantReadWriteLock modelLock

queue

protected final BlockingQueue<Model> queue

isCG

protected transient boolean isCG

isMLN

protected transient boolean isMLN

Constructor Detail

ModelHolder

protected ModelHolder()

Method Detail

init

protected void init()

acquireModel

protected Model acquireModel()
                      throws InterruptedException

Throws:

InterruptedException

releaseModel

protected void releaseModel(Model model)

output

protected INDArray[] output(INDArray[] input,
                            INDArray[] inputMasks)

updateModel

protected void updateModel(@NonNull
                           @NonNull Model model)