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

Class CPUTransformAlongDimensionTask

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.transform.CPUTransformAlongDimensionTask

All Implemented Interfaces:

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

public class CPUTransformAlongDimensionTask
extends BaseCPUAction

See Also:

Serialized Form

Field Summary

Fields

Modifier and Type	Field and Description
protected int[]	dimensions
protected TransformOp	op
protected List<Task<Void>>	subTasks

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

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

Constructor Summary

Constructors

Constructor and Description
CPUTransformAlongDimensionTask(TransformOp op, int threshold, int... dimensions)

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

Field Detail

op

protected final TransformOp op

dimensions

protected final int[] dimensions

subTasks

protected List<Task<Void>> subTasks

Constructor Detail

CPUTransformAlongDimensionTask

public CPUTransformAlongDimensionTask(TransformOp op,
                                      int threshold,
                                      int... dimensions)

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