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

Class CPUAccumulationViaTensorTask

java.lang.Object

java.util.concurrent.ForkJoinTask<V>

java.util.concurrent.RecursiveTask<V>

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

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

All Implemented Interfaces:

Serializable, Callable<Double>, Future<Double>, Task<Double>

public class CPUAccumulationViaTensorTask
extends BaseCPUTask<Double>

See Also:

Serialized Form

Field Summary

Fields

Modifier and Type	Field and Description
protected Accumulation	op
protected boolean	outerTask
protected List<Task<Double>>	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
CPUAccumulationViaTensorTask(Accumulation op, int threshold, boolean outerTask)

Method Summary

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

outerTask

protected final boolean outerTask

subTasks

protected List<Task<Double>> subTasks

Constructor Detail

CPUAccumulationViaTensorTask

public CPUAccumulationViaTensorTask(Accumulation op,
                                    int threshold,
                                    boolean outerTask)

Method Detail

blockUntilComplete

public Double blockUntilComplete()

Description copied from interface: Task

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

call

public Double call()

Description copied from interface: Task

Compute the result immediately, in the current thread

compute

protected Double compute()

Specified by:

compute in class RecursiveTask<Double>