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

Class CPUIndexAccumulations1dAction

java.lang.Object

java.util.concurrent.ForkJoinTask<Void>

java.util.concurrent.RecursiveAction

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

All Implemented Interfaces:

Serializable, Callable<Void>, Future<Void>, Task<Void>

public class CPUIndexAccumulations1dAction
extends RecursiveAction
implements Task<Void>

A class used specifically for IndexAccumulation along dimensions for 1d. It differs from other IndexAccumulation tasks in that it assigns the output directly instead of returning, plus does multiple index accumulation ops in the one object (helps to avoid object creation cost for large number of small ops)

See Also:

Serialized Form

Constructor Summary

Constructors

Constructor and Description
CPUIndexAccumulations1dAction(IndexAccumulation op, int threshold, TensorCalculator tCalcx, TensorCalculator tCalcy, int firstTensor, int lastTensor, INDArray output)

Method Summary

Modifier and Type	Method and Description
Void	blockUntilComplete() Assuming invokeAsync() has been called, block until the execution completes
Void	call() Compute the result immediately, in the current thread
protected void	compute()
void	invokeAsync() Schedule for asyncronous execution; returns immediately
Void	invokeBlocking() Schedule for execution, and block until completion

Methods inherited from class java.util.concurrent.RecursiveAction

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

Constructor Detail

CPUIndexAccumulations1dAction

public CPUIndexAccumulations1dAction(IndexAccumulation op,
                                     int threshold,
                                     TensorCalculator tCalcx,
                                     TensorCalculator tCalcy,
                                     int firstTensor,
                                     int lastTensor,
                                     INDArray output)

Method Detail

compute

protected void compute()

Specified by:

compute in class RecursiveAction

call

public Void call()

Description copied from interface: Task

Compute the result immediately, in the current thread

Specified by:

call in interface Callable<Void>

Specified by:

call in interface Task<Void>

invokeBlocking

public Void invokeBlocking()

Description copied from interface: Task

Schedule for execution, and block until completion

Specified by:

invokeBlocking in interface Task<Void>

invokeAsync

public void invokeAsync()

Description copied from interface: Task

Schedule for asyncronous execution; returns immediately

Specified by:

invokeAsync in interface Task<Void>

blockUntilComplete

public Void blockUntilComplete()

Description copied from interface: Task

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

Specified by:

blockUntilComplete in interface Task<Void>