org.nd4j.linalg.api.parallel

Class TaskCreator

java.lang.Object

org.nd4j.linalg.api.parallel.TaskCreator

public class TaskCreator
extends Object

Created by agibsonccc on 9/19/15.

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
static class	TaskCreator.AccumulationINDArrayTask
static class	TaskCreator.ForkJoinArrayINDArrayTask
static class	TaskCreator.ForkJoinINDArrayTask
static interface	TaskCreator.INDArrayTask
static class	TaskCreator.OpINDArrayTask
static class	TaskCreator.RunnableINDArrayTask
static class	TaskCreator.RunnableMultipleINDArrayTask

Constructor Summary

Constructors

Constructor and Description
TaskCreator()

Method Summary

Modifier and Type	Method and Description
static org.apache.commons.math3.util.Pair<List<ForkJoinTask<INDArray[]>>,CountDownLatch>	parititonForkJoinBasedOnSlices(INDArray[] arr, TaskCreator.INDArrayTask task)
static org.apache.commons.math3.util.Pair<List<ForkJoinTask<INDArray>>,CountDownLatch>	parititonForkJoinBasedOnSlices(INDArray arr, Op op, OpExecutioner opExecutioner)
static org.apache.commons.math3.util.Pair<List<ForkJoinTask<INDArray>>,CountDownLatch>	parititonForkJoinBasedOnSlices(INDArray arr, TaskCreator.INDArrayTask task)
static org.apache.commons.math3.util.Pair<List<ForkJoinTask<INDArray[]>>,CountDownLatch>	parititonForkJoinBasedOnTensorsAlongDimension(INDArray[] arr, Op op, OpExecutioner opExecutioner, int... dimension)
static org.apache.commons.math3.util.Pair<List<ForkJoinTask<INDArray[]>>,CountDownLatch>	parititonForkJoinBasedOnTensorsAlongDimension(INDArray[] arr, TaskCreator.INDArrayTask task, int... dimension)
static List<ForkJoinTask<INDArray>>	parititonForkJoinBasedOnTensorsAlongDimension(INDArray arr, Accumulation op, OpExecutioner opExecutioner, INDArray retArray, int... dimension)
static org.apache.commons.math3.util.Pair<CountDownLatch,List<ForkJoinTask<INDArray>>>	parititonForkJoinBasedOnTensorsAlongDimension(INDArray arr, Op op, OpExecutioner opExecutioner, int... dimension)
static org.apache.commons.math3.util.Pair<List<ForkJoinTask<INDArray>>,CountDownLatch>	parititonForkJoinBasedOnTensorsAlongDimension(INDArray arr, TaskCreator.INDArrayTask task, int... dimension)
static org.apache.commons.math3.util.Pair<List<Runnable>,CountDownLatch>	parititonRunnablesBasedOnSlices(INDArray[] arr, TaskCreator.INDArrayTask task)
static org.apache.commons.math3.util.Pair<List<Runnable>,CountDownLatch>	parititonRunnablesBasedOnSlices(INDArray arr, Op op, OpExecutioner opExecutioner)
static org.apache.commons.math3.util.Pair<List<Runnable>,CountDownLatch>	parititonRunnablesBasedOnSlices(INDArray arr, TaskCreator.INDArrayTask task)
static List<Runnable>	parititonRunnablesBasedOnTensorsAlongDimension(INDArray[] arr, TaskCreator.INDArrayTask task, int... dimension)
static org.apache.commons.math3.util.Pair<List<Runnable>,CountDownLatch>	parititonRunnablesBasedOnTensorsAlongDimension(INDArray arr, Op task, OpExecutioner opExecutioner, int... dimension)
static List<Runnable>	parititonRunnablesBasedOnTensorsAlongDimension(INDArray arr, TaskCreator.INDArrayTask task, int... dimension)

Methods inherited from class java.lang.Object

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

Constructor Detail

TaskCreator

public TaskCreator()

Method Detail

parititonForkJoinBasedOnSlices

public static org.apache.commons.math3.util.Pair<List<ForkJoinTask<INDArray>>,CountDownLatch> parititonForkJoinBasedOnSlices(INDArray arr,
                                                                                                                             Op op,
                                                                                                                             OpExecutioner opExecutioner)

Parameters:

arr -

op -

opExecutioner -

Returns:

parititonRunnablesBasedOnSlices

public static org.apache.commons.math3.util.Pair<List<Runnable>,CountDownLatch> parititonRunnablesBasedOnSlices(INDArray arr,
                                                                                                                Op op,
                                                                                                                OpExecutioner opExecutioner)

Parameters:

arr -

op -

opExecutioner -

Returns:

parititonForkJoinBasedOnTensorsAlongDimension

public static org.apache.commons.math3.util.Pair<List<ForkJoinTask<INDArray[]>>,CountDownLatch> parititonForkJoinBasedOnTensorsAlongDimension(INDArray[] arr,
                                                                                                                                              Op op,
                                                                                                                                              OpExecutioner opExecutioner,
                                                                                                                                              int... dimension)

Parameters:

arr -

op -

opExecutioner -

dimension -

Returns:

parititonRunnablesBasedOnTensorsAlongDimension

public static List<Runnable> parititonRunnablesBasedOnTensorsAlongDimension(INDArray[] arr,
                                                                            TaskCreator.INDArrayTask task,
                                                                            int... dimension)

Parameters:

arr -

task -

dimension -

Returns:

parititonForkJoinBasedOnTensorsAlongDimension

public static List<ForkJoinTask<INDArray>> parititonForkJoinBasedOnTensorsAlongDimension(INDArray arr,
                                                                                         Accumulation op,
                                                                                         OpExecutioner opExecutioner,
                                                                                         INDArray retArray,
                                                                                         int... dimension)

Parameters:

arr -

op -

opExecutioner -

dimension -

Returns:

parititonForkJoinBasedOnTensorsAlongDimension

public static org.apache.commons.math3.util.Pair<CountDownLatch,List<ForkJoinTask<INDArray>>> parititonForkJoinBasedOnTensorsAlongDimension(INDArray arr,
                                                                                                                                            Op op,
                                                                                                                                            OpExecutioner opExecutioner,
                                                                                                                                            int... dimension)

Parameters:

arr -

op -

opExecutioner -

dimension -

Returns:

parititonRunnablesBasedOnTensorsAlongDimension

public static List<Runnable> parititonRunnablesBasedOnTensorsAlongDimension(INDArray arr,
                                                                            TaskCreator.INDArrayTask task,
                                                                            int... dimension)

Parameters:

arr -

task -

dimension -

Returns:

parititonForkJoinBasedOnSlices

public static org.apache.commons.math3.util.Pair<List<ForkJoinTask<INDArray[]>>,CountDownLatch> parititonForkJoinBasedOnSlices(INDArray[] arr,
                                                                                                                               TaskCreator.INDArrayTask task)

Parameters:

arr -

task -

Returns:

parititonRunnablesBasedOnSlices

public static org.apache.commons.math3.util.Pair<List<Runnable>,CountDownLatch> parititonRunnablesBasedOnSlices(INDArray[] arr,
                                                                                                                TaskCreator.INDArrayTask task)

Parameters:

arr -

task -

Returns:

parititonForkJoinBasedOnSlices

public static org.apache.commons.math3.util.Pair<List<ForkJoinTask<INDArray>>,CountDownLatch> parititonForkJoinBasedOnSlices(INDArray arr,
                                                                                                                             TaskCreator.INDArrayTask task)

Parameters:

arr -

task -

Returns:

parititonRunnablesBasedOnSlices

public static org.apache.commons.math3.util.Pair<List<Runnable>,CountDownLatch> parititonRunnablesBasedOnSlices(INDArray arr,
                                                                                                                TaskCreator.INDArrayTask task)

Parameters:

arr -

task -

Returns:

parititonForkJoinBasedOnTensorsAlongDimension

public static org.apache.commons.math3.util.Pair<List<ForkJoinTask<INDArray>>,CountDownLatch> parititonForkJoinBasedOnTensorsAlongDimension(INDArray arr,
                                                                                                                                            TaskCreator.INDArrayTask task,
                                                                                                                                            int... dimension)

Parameters:

arr -

task -

dimension -

Returns:

parititonRunnablesBasedOnTensorsAlongDimension

public static org.apache.commons.math3.util.Pair<List<Runnable>,CountDownLatch> parititonRunnablesBasedOnTensorsAlongDimension(INDArray arr,
                                                                                                                               Op task,
                                                                                                                               OpExecutioner opExecutioner,
                                                                                                                               int... dimension)

Parameters:

arr -

task -

dimension -

Returns:

parititonForkJoinBasedOnTensorsAlongDimension

public static org.apache.commons.math3.util.Pair<List<ForkJoinTask<INDArray[]>>,CountDownLatch> parititonForkJoinBasedOnTensorsAlongDimension(INDArray[] arr,
                                                                                                                                              TaskCreator.INDArrayTask task,
                                                                                                                                              int... dimension)

Parameters:

arr -

task -

dimension -

Returns: