org.nd4j.linalg.api.parallel.tasks.cpu

Class CPUTaskFactory

java.lang.Object

org.nd4j.linalg.api.parallel.tasks.cpu.CPUTaskFactory

All Implemented Interfaces:

TaskFactory

public class CPUTaskFactory
extends Object
implements TaskFactory

TaskFactory for CPU backends

Field Summary

Fields

Modifier and Type	Field and Description
static String	PARALLEL_THRESHOLD
protected int	parallelThreshold

Constructor Summary

Constructors

Constructor and Description
CPUTaskFactory()

Method Summary

Modifier and Type	Method and Description
Task<Double>	getAccumulationTask(Accumulation op)
Task<Double>	getAccumulationTask(Accumulation op, boolean outerTask)
Task<INDArray>	getAccumulationTask(Accumulation op, int... dimension)
Task<Void>	getBroadcastOpAction(BroadcastOp op)
Task<INDArray>	getCol2ImTask(INDArray col, int strideY, int strideX, int padHeight, int padWidth, int imgHeight, int imgWidth)
Task<INDArray>	getIm2ColTask(INDArray img, int kernelHeight, int kernelWidth, int strideY, int strideX, int padHeight, int padWidth, boolean coverAll)
Task<org.apache.commons.math3.util.Pair<Double,Integer>>	getIndexAccumulationTask(IndexAccumulation op)
Task<INDArray>	getIndexAccumulationTask(IndexAccumulation op, int... dimension)
int	getParallelThreshold() Get the current threshold for parallel execution of tasks
Task<Void>	getScalarAction(ScalarOp op)
Task<Void>	getTransformAction(TransformOp op)
Task<Void>	getTransformAction(TransformOp op, int... dimension)
void	setParallelThreshold(int threshold) Set the threshold for breaking up CPU tasks for parallel execution.

Methods inherited from class java.lang.Object

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

Field Detail

PARALLEL_THRESHOLD

public static final String PARALLEL_THRESHOLD

See Also:

Constant Field Values

parallelThreshold

protected int parallelThreshold

Constructor Detail

CPUTaskFactory

public CPUTaskFactory()

Method Detail

setParallelThreshold

public void setParallelThreshold(int threshold)

Set the threshold for breaking up CPU tasks for parallel execution. If a given task/op has length greater than the threshold, it will be broken down into a number of smaller tasks for execution.

Parameters:

threshold - New threshold to use for parallel execution

getParallelThreshold

public int getParallelThreshold()

Get the current threshold for parallel execution of tasks

getTransformAction

public Task<Void> getTransformAction(TransformOp op)

Specified by:

getTransformAction in interface TaskFactory

getTransformAction

public Task<Void> getTransformAction(TransformOp op,
                                     int... dimension)

Specified by:

getTransformAction in interface TaskFactory

getScalarAction

public Task<Void> getScalarAction(ScalarOp op)

Specified by:

getScalarAction in interface TaskFactory

getAccumulationTask

public Task<Double> getAccumulationTask(Accumulation op,
                                        boolean outerTask)

Specified by:

getAccumulationTask in interface TaskFactory

getAccumulationTask

public Task<Double> getAccumulationTask(Accumulation op)

Specified by:

getAccumulationTask in interface TaskFactory

getAccumulationTask

public Task<INDArray> getAccumulationTask(Accumulation op,
                                          int... dimension)

Specified by:

getAccumulationTask in interface TaskFactory

getIndexAccumulationTask

public Task<org.apache.commons.math3.util.Pair<Double,Integer>> getIndexAccumulationTask(IndexAccumulation op)

Specified by:

getIndexAccumulationTask in interface TaskFactory

getIndexAccumulationTask

public Task<INDArray> getIndexAccumulationTask(IndexAccumulation op,
                                               int... dimension)

Specified by:

getIndexAccumulationTask in interface TaskFactory

getBroadcastOpAction

public Task<Void> getBroadcastOpAction(BroadcastOp op)

Specified by:

getBroadcastOpAction in interface TaskFactory

getIm2ColTask

public Task<INDArray> getIm2ColTask(INDArray img,
                                    int kernelHeight,
                                    int kernelWidth,
                                    int strideY,
                                    int strideX,
                                    int padHeight,
                                    int padWidth,
                                    boolean coverAll)

Specified by:

getIm2ColTask in interface TaskFactory

getCol2ImTask

public Task<INDArray> getCol2ImTask(INDArray col,
                                    int strideY,
                                    int strideX,
                                    int padHeight,
                                    int padWidth,
                                    int imgHeight,
                                    int imgWidth)

Specified by:

getCol2ImTask in interface TaskFactory