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

Class CPUScalarOpAction

java.lang.Object

java.util.concurrent.ForkJoinTask<Void>

java.util.concurrent.RecursiveAction

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

org.nd4j.linalg.api.parallel.tasks.cpu.scalar.BaseCPUScalarOpAction

org.nd4j.linalg.api.parallel.tasks.cpu.scalar.CPUScalarOpAction

All Implemented Interfaces:

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

public class CPUScalarOpAction
extends BaseCPUScalarOpAction

See Also:

Serialized Form

Field Summary

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

op

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

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

Constructor Summary

Constructors

Constructor and Description
CPUScalarOpAction(ScalarOp op, int threshold) Constructor for doing task on entire NDArray
CPUScalarOpAction(ScalarOp op, int threshold, int tadIdx, int tadDim) Constructor for doing a 1d tensor first
CPUScalarOpAction(ScalarOp op, int threshold, int n, int offsetX, int offsetZ, int incrX, int incrZ) Constructor for operating on subset of NDArray

Method Summary

Modifier and Type	Method and Description
Void	call() Compute the result immediately, in the current thread
protected void	compute()

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

blockUntilComplete, doTensorFirst, invokeAsync, invokeBlocking

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

CPUScalarOpAction

public CPUScalarOpAction(ScalarOp op,
                         int threshold,
                         int n,
                         int offsetX,
                         int offsetZ,
                         int incrX,
                         int incrZ)

Constructor for operating on subset of NDArray

CPUScalarOpAction

public CPUScalarOpAction(ScalarOp op,
                         int threshold)

Constructor for doing task on entire NDArray

CPUScalarOpAction

public CPUScalarOpAction(ScalarOp op,
                         int threshold,
                         int tadIdx,
                         int tadDim)

Constructor for doing a 1d tensor first

Method Detail

call

public Void call()

Description copied from interface: Task

Compute the result immediately, in the current thread

compute

protected void compute()

Specified by:

compute in class RecursiveAction