org.nd4j.linalg.api.parallel

Class DefaultParallelExecutioner

java.lang.Object

org.nd4j.linalg.api.parallel.DefaultParallelExecutioner

All Implemented Interfaces:

ParallelExecutioner

public class DefaultParallelExecutioner
extends Object
implements ParallelExecutioner

Parallel executioner

Author:

Adam Gibson

Constructor Summary

Constructors

Constructor and Description
DefaultParallelExecutioner()
DefaultParallelExecutioner(ExecutorService executorService)
DefaultParallelExecutioner(ForkJoinPool forkJoinPool)

Method Summary

Modifier and Type	Method and Description
<T> void	exec(ForkJoinTask<T> task)
Future	exec(Runnable runnable)
void	execBasedOnArraysAlongDimension(INDArray[] arr, TaskCreator.INDArrayTask task, int... dimension)
INDArray	execBasedOnArraysAlongDimension(INDArray arr, Accumulation task, OpExecutioner executioner, int... dimension)
void	execBasedOnArraysAlongDimension(INDArray arr, Op task, OpExecutioner executioner, int... dimension)
void	execBasedOnArraysAlongDimension(INDArray arr, TaskCreator.INDArrayTask task, int... dimension)
void	execBasedOnSlices(INDArray arr, Op task, OpExecutioner executioner)
void	execBasedOnSlices(INDArray arr, TaskCreator.INDArrayTask task)

Methods inherited from class java.lang.Object

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

Constructor Detail

DefaultParallelExecutioner

public DefaultParallelExecutioner(ForkJoinPool forkJoinPool)

DefaultParallelExecutioner

public DefaultParallelExecutioner(ExecutorService executorService)

DefaultParallelExecutioner

public DefaultParallelExecutioner()

Method Detail

execBasedOnArraysAlongDimension

public INDArray execBasedOnArraysAlongDimension(INDArray arr,
                                                Accumulation task,
                                                OpExecutioner executioner,
                                                int... dimension)

Specified by:

execBasedOnArraysAlongDimension in interface ParallelExecutioner

execBasedOnArraysAlongDimension

public void execBasedOnArraysAlongDimension(INDArray arr,
                                            Op task,
                                            OpExecutioner executioner,
                                            int... dimension)

Specified by:

execBasedOnArraysAlongDimension in interface ParallelExecutioner

execBasedOnSlices

public void execBasedOnSlices(INDArray arr,
                              Op task,
                              OpExecutioner executioner)

Specified by:

execBasedOnSlices in interface ParallelExecutioner

execBasedOnArraysAlongDimension

public void execBasedOnArraysAlongDimension(INDArray arr,
                                            TaskCreator.INDArrayTask task,
                                            int... dimension)

Specified by:

execBasedOnArraysAlongDimension in interface ParallelExecutioner

execBasedOnArraysAlongDimension

public void execBasedOnArraysAlongDimension(INDArray[] arr,
                                            TaskCreator.INDArrayTask task,
                                            int... dimension)

Specified by:

execBasedOnArraysAlongDimension in interface ParallelExecutioner

execBasedOnSlices

public void execBasedOnSlices(INDArray arr,
                              TaskCreator.INDArrayTask task)

Specified by:

execBasedOnSlices in interface ParallelExecutioner

exec

public Future exec(Runnable runnable)

Specified by:

exec in interface ParallelExecutioner

Returns:

exec

public <T> void exec(ForkJoinTask<T> task)

Specified by:

exec in interface ParallelExecutioner