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

Class BaseCPUAction

java.lang.Object

java.util.concurrent.ForkJoinTask<Void>

java.util.concurrent.RecursiveAction

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

All Implemented Interfaces:

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

Direct Known Subclasses:

BaseCPUScalarOpAction, BaseCPUTransformOpAction, CPUScalarOpViaTensorAction, CPUTransformAlongDimensionTask, CPUTransformOpViaTensorTask, CPUVectorOp

public abstract class BaseCPUAction
extends RecursiveAction
implements Task<Void>

See Also:

Serialized Form

Field Summary

Fields

Modifier and Type	Field and Description
protected boolean	doTensorFirst
protected boolean	executed
protected Future<Void>	future
protected int	incrX
protected int	incrY
protected int	incrZ
protected int	n
protected int	offsetX
protected int	offsetY
protected int	offsetZ
protected List<Task<Void>>	subTasks
protected int	tensorDim
protected int	tensorIdx
protected int	threshold

Constructor Summary

Constructors

Constructor and Description
BaseCPUAction(int threshold, int n, int offsetX, int offsetY, int offsetZ, int incrX, int incrY, int incrZ)
BaseCPUAction(Op op, int threshold)
BaseCPUAction(Op op, int threshold, int tadIdx, int tadDim) Constructor for doing a 1d tensor along dimension first

Method Summary

Modifier and Type	Method and Description
Void	blockUntilComplete() Assuming invokeAsync() has been called, block until the execution completes
protected void	doTensorFirst(Op op)
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

compute, 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

Methods inherited from interface org.nd4j.linalg.api.parallel.tasks.Task

call

Field Detail

threshold

protected final int threshold

n

protected int n

offsetX

protected int offsetX

offsetY

protected int offsetY

offsetZ

protected int offsetZ

incrX

protected int incrX

incrY

protected int incrY

incrZ

protected int incrZ

doTensorFirst

protected boolean doTensorFirst

tensorIdx

protected int tensorIdx

tensorDim

protected int tensorDim

executed

protected boolean executed

future

protected Future<Void> future

subTasks

protected List<Task<Void>> subTasks

Constructor Detail

BaseCPUAction

public BaseCPUAction(int threshold,
                     int n,
                     int offsetX,
                     int offsetY,
                     int offsetZ,
                     int incrX,
                     int incrY,
                     int incrZ)

BaseCPUAction

public BaseCPUAction(Op op,
                     int threshold)

BaseCPUAction

public BaseCPUAction(Op op,
                     int threshold,
                     int tadIdx,
                     int tadDim)

Constructor for doing a 1d tensor along dimension first

Method Detail

doTensorFirst

protected void doTensorFirst(Op op)

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>

invokeBlocking

public Void invokeBlocking()

Description copied from interface: Task

Schedule for execution, and block until completion

Specified by:

invokeBlocking in interface Task<Void>