org.deeplearning4j.parallelism

Class ParallelInference

java.lang.Object

org.deeplearning4j.parallelism.ParallelInference

public class ParallelInference
extends Object

This class is simple wrapper for ParallelInference using batched input

Author:

raver119@gmail.com

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
static class	ParallelInference.Builder
protected static class	ParallelInference.ObservablesProvider

Field Summary

Fields

Modifier and Type	Field and Description
static int	DEFAULT_BATCH_LIMIT
static InferenceMode	DEFAULT_INFERENCE_MODE
static int	DEFAULT_NUM_WORKERS
static int	DEFAULT_QUEUE_LIMIT

Constructor Summary

Constructors

Modifier	Constructor and Description
protected	ParallelInference()

Method Summary

Modifier and Type	Method and Description
protected long	getWorkerCounter(int workerIdx)
protected void	init()
org.nd4j.linalg.api.ndarray.INDArray	output(org.nd4j.linalg.dataset.DataSet dataSet)
org.nd4j.linalg.api.ndarray.INDArray	output(double[] input)
org.nd4j.linalg.api.ndarray.INDArray	output(float[] input)
org.nd4j.linalg.api.ndarray.INDArray[]	output(org.nd4j.linalg.api.ndarray.INDArray... input)
org.nd4j.linalg.api.ndarray.INDArray	output(org.nd4j.linalg.api.ndarray.INDArray input)

Methods inherited from class java.lang.Object

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

Field Detail

DEFAULT_NUM_WORKERS

public static final int DEFAULT_NUM_WORKERS

DEFAULT_BATCH_LIMIT

public static final int DEFAULT_BATCH_LIMIT

See Also:

Constant Field Values

DEFAULT_INFERENCE_MODE

public static final InferenceMode DEFAULT_INFERENCE_MODE

DEFAULT_QUEUE_LIMIT

public static final int DEFAULT_QUEUE_LIMIT

See Also:

Constant Field Values

Constructor Detail

ParallelInference

protected ParallelInference()

Method Detail

init

protected void init()

getWorkerCounter

protected long getWorkerCounter(int workerIdx)

output

public org.nd4j.linalg.api.ndarray.INDArray output(double[] input)

Parameters:

input -

Returns:

output

public org.nd4j.linalg.api.ndarray.INDArray output(float[] input)

Parameters:

input -

Returns:

output

public org.nd4j.linalg.api.ndarray.INDArray output(org.nd4j.linalg.api.ndarray.INDArray input)

output

public org.nd4j.linalg.api.ndarray.INDArray output(org.nd4j.linalg.dataset.DataSet dataSet)

Parameters:

dataSet -

Returns:

output

public org.nd4j.linalg.api.ndarray.INDArray[] output(org.nd4j.linalg.api.ndarray.INDArray... input)

Parameters:

input -

Returns: