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

Class CPUIndexAccumulationAlongDimensionTask

java.lang.Object

java.util.concurrent.ForkJoinTask<V>

java.util.concurrent.RecursiveTask<V>

org.nd4j.linalg.api.parallel.tasks.cpu.BaseCPUTask<INDArray>

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

All Implemented Interfaces:

Serializable, Callable<INDArray>, Future<INDArray>, Task<INDArray>

public class CPUIndexAccumulationAlongDimensionTask
extends BaseCPUTask<INDArray>

See Also:

Serialized Form

Field Summary

Fields

Modifier and Type	Field and Description
protected int[]	dimensions
protected IndexAccumulation	op
protected List<Task<org.apache.commons.math3.util.Pair<Double,Integer>>>	subTasks

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
CPUIndexAccumulationAlongDimensionTask(IndexAccumulation op, int parallelThreshold, int[] dimensions)

Method Summary

Modifier and Type	Method and Description
INDArray	blockUntilComplete() Assuming invokeAsync() has been called, block until the execution completes
INDArray	call() Compute the result immediately, in the current thread
protected INDArray	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

op

protected final IndexAccumulation op

dimensions

protected final int[] dimensions

subTasks

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

Constructor Detail

CPUIndexAccumulationAlongDimensionTask

public CPUIndexAccumulationAlongDimensionTask(IndexAccumulation op,
                                              int parallelThreshold,
                                              int[] dimensions)

Method Detail

blockUntilComplete

public INDArray blockUntilComplete()

Description copied from interface: Task

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

call

public INDArray call()

Description copied from interface: Task

Compute the result immediately, in the current thread

compute

protected INDArray compute()

Specified by:

compute in class RecursiveTask<INDArray>