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

Class CPUIndexAccumulationTask

java.lang.Object

java.util.concurrent.ForkJoinTask<V>

java.util.concurrent.RecursiveTask<V>

org.nd4j.linalg.api.parallel.tasks.cpu.BaseCPUTask<org.apache.commons.math3.util.Pair<Double,Integer>>

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

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

All Implemented Interfaces:

Serializable, Callable<org.apache.commons.math3.util.Pair<Double,Integer>>, Future<org.apache.commons.math3.util.Pair<Double,Integer>>, Task<org.apache.commons.math3.util.Pair<Double,Integer>>

public class CPUIndexAccumulationTask
extends BaseCPUIndexAccumulationTask

See Also:

Serialized Form

Field Summary

Fields

Modifier and Type	Field and Description
protected List<Task<org.apache.commons.math3.util.Pair<Double,Integer>>>	subTasks

Fields inherited from class org.nd4j.linalg.api.parallel.tasks.cpu.indexaccum.BaseCPUIndexAccumulationTask

elementOffset, op, outerTask

Fields inherited from class org.nd4j.linalg.api.parallel.tasks.cpu.BaseCPUTask

doTensorFirst, executed, future, incrX, incrY, incrZ, n, offsetX, offsetY, offsetZ, tensorDim, tensorIdx, threshold

Constructor Summary

Constructors

Constructor and Description
CPUIndexAccumulationTask(IndexAccumulation op, int threshold, boolean outerTask) Constructor for doing task on entire NDArray
CPUIndexAccumulationTask(IndexAccumulation op, int threshold, int tadIdx, int tadDim, boolean outerTask) Constructor for doing a 1d tensor first
CPUIndexAccumulationTask(IndexAccumulation op, int threshold, int n, int offsetX, int offsetY, int incrX, int incrY, int elementOffset, boolean outerTask) Constructor for operating on subset of NDArray

Method Summary

Modifier and Type	Method and Description
org.apache.commons.math3.util.Pair<Double,Integer>	blockUntilComplete() Assuming invokeAsync() has been called, block until the execution completes
org.apache.commons.math3.util.Pair<Double,Integer>	call() Compute the result immediately, in the current thread
protected org.apache.commons.math3.util.Pair<Double,Integer>	compute()

Methods inherited from class org.nd4j.linalg.api.parallel.tasks.cpu.BaseCPUTask

doTensorFirst, invokeAsync, invokeBlocking

Methods inherited from class java.util.concurrent.RecursiveTask

exec, getRawResult, setRawResult

Methods inherited from class java.util.concurrent.ForkJoinTask

adapt, adapt, adapt, cancel, compareAndSetForkJoinTaskTag, complete, completeExceptionally, fork, get, get, getException, getForkJoinTaskTag, getPool, getQueuedTaskCount, getSurplusQueuedTaskCount, helpQuiesce, inForkJoinPool, invoke, invokeAll, invokeAll, invokeAll, isCancelled, isCompletedAbnormally, isCompletedNormally, isDone, join, peekNextLocalTask, pollNextLocalTask, pollTask, quietlyComplete, quietlyInvoke, quietlyJoin, reinitialize, setForkJoinTaskTag, tryUnfork

Methods inherited from class java.lang.Object

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

Field Detail

subTasks

protected List<Task<org.apache.commons.math3.util.Pair<Double,Integer>>> subTasks

Constructor Detail

CPUIndexAccumulationTask

public CPUIndexAccumulationTask(IndexAccumulation op,
                                int threshold,
                                int n,
                                int offsetX,
                                int offsetY,
                                int incrX,
                                int incrY,
                                int elementOffset,
                                boolean outerTask)

Constructor for operating on subset of NDArray

CPUIndexAccumulationTask

public CPUIndexAccumulationTask(IndexAccumulation op,
                                int threshold,
                                boolean outerTask)

Constructor for doing task on entire NDArray

CPUIndexAccumulationTask

public CPUIndexAccumulationTask(IndexAccumulation op,
                                int threshold,
                                int tadIdx,
                                int tadDim,
                                boolean outerTask)

Constructor for doing a 1d tensor first

Method Detail

blockUntilComplete

public org.apache.commons.math3.util.Pair<Double,Integer> blockUntilComplete()

Description copied from interface: Task

Assuming invokeAsync() has been called, block until the execution completes

call

public org.apache.commons.math3.util.Pair<Double,Integer> call()

Description copied from interface: Task

Compute the result immediately, in the current thread

compute

protected org.apache.commons.math3.util.Pair<Double,Integer> compute()

Specified by:

compute in class RecursiveTask<org.apache.commons.math3.util.Pair<Double,Integer>>