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

Class BaseCPUIndexAccumulationTask

java.lang.Object

java.util.concurrent.ForkJoinTask<V>

java.util.concurrent.RecursiveTask<V>

org.nd4j.linalg.api.parallel.tasks.cpu.BaseCPUTask<org.apache.commons.math3.util.Pair<Double,Integer>>

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

All Implemented Interfaces:

Serializable, Callable<org.apache.commons.math3.util.Pair<Double,Integer>>, Future<org.apache.commons.math3.util.Pair<Double,Integer>>, Task<org.apache.commons.math3.util.Pair<Double,Integer>>

Direct Known Subclasses:

CPUIndexAccumulationTask

public abstract class BaseCPUIndexAccumulationTask
extends BaseCPUTask<org.apache.commons.math3.util.Pair<Double,Integer>>

See Also:

Serialized Form

Field Summary

Fields

Modifier and Type	Field and Description
protected int	elementOffset
protected IndexAccumulation	op
protected boolean	outerTask

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
BaseCPUIndexAccumulationTask(IndexAccumulation op, int threshold, boolean outerTask) Constructor for doing task on entire NDArray
BaseCPUIndexAccumulationTask(IndexAccumulation op, int threshold, int tadIdx, int tadDim, boolean outerTask) Constructor for doing a 1d tensor first
BaseCPUIndexAccumulationTask(IndexAccumulation op, int threshold, int n, int offsetX, int offsetY, int incrX, int incrY, int elementOffset, boolean outerTask) Constructor for operating on subset of NDArray

Method Summary

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

doTensorFirst, invokeAsync, invokeBlocking

Methods inherited from class java.util.concurrent.RecursiveTask

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

blockUntilComplete, call

Field Detail

op

protected final IndexAccumulation op

elementOffset

protected final int elementOffset

outerTask

protected final boolean outerTask

Constructor Detail

BaseCPUIndexAccumulationTask

public BaseCPUIndexAccumulationTask(IndexAccumulation op,
                                    int threshold,
                                    int n,
                                    int offsetX,
                                    int offsetY,
                                    int incrX,
                                    int incrY,
                                    int elementOffset,
                                    boolean outerTask)

Constructor for operating on subset of NDArray

BaseCPUIndexAccumulationTask

public BaseCPUIndexAccumulationTask(IndexAccumulation op,
                                    int threshold,
                                    boolean outerTask)

Constructor for doing task on entire NDArray

BaseCPUIndexAccumulationTask

public BaseCPUIndexAccumulationTask(IndexAccumulation op,
                                    int threshold,
                                    int tadIdx,
                                    int tadDim,
                                    boolean outerTask)

Constructor for doing a 1d tensor first