public class CPUIm2ColTask extends RecursiveTask<INDArray> implements Task<INDArray>
Modifier and Type | Field and Description |
---|---|
protected boolean |
coverAll |
protected int |
depthFrom |
protected int |
depthTo |
protected int |
exampleFrom |
protected int |
exampleTo |
protected Future<INDArray> |
future |
protected INDArray |
img |
protected int |
kernelHeight |
protected int |
kernelWidth |
protected INDArray |
out |
protected int |
padHeight |
protected int |
padWidth |
protected int |
parallelThreshold |
protected int |
strideX |
protected int |
strideY |
protected List<CPUIm2ColTask> |
subTasks |
protected int |
xOutFrom |
protected int |
xOutTo |
protected int |
yOutFrom |
protected int |
yOutTo |
Constructor and Description |
---|
CPUIm2ColTask(INDArray img,
INDArray out,
int kernelHeight,
int kernelWidth,
int strideY,
int strideX,
int padHeight,
int padWidth,
int exampleFrom,
int exampleTo,
int depthFrom,
int depthTo,
int yOutFrom,
int yOutTo,
int xOutFrom,
int xOutTo,
boolean coverAll,
int parallelThreshold) |
CPUIm2ColTask(INDArray img,
int kernelHeight,
int kernelWidth,
int strideY,
int strideX,
int padHeight,
int padWidth,
boolean coverAll,
int parallelThreshold) |
Modifier and Type | Method and Description |
---|---|
INDArray |
blockUntilComplete()
Assuming invokeAsync() has been called, block until the execution completes
|
INDArray |
call()
Compute the result immediately, in the current thread
|
protected INDArray |
compute() |
void |
invokeAsync()
Schedule for asyncronous execution; returns immediately
|
INDArray |
invokeBlocking()
Schedule for execution, and block until completion
|
exec, getRawResult, setRawResult
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
protected List<CPUIm2ColTask> subTasks
protected final INDArray img
protected INDArray out
protected final int kernelHeight
protected final int kernelWidth
protected final int strideY
protected final int strideX
protected final int padHeight
protected final int padWidth
protected final boolean coverAll
protected final int parallelThreshold
protected final int exampleFrom
protected final int exampleTo
protected final int depthFrom
protected final int depthTo
protected final int xOutFrom
protected final int xOutTo
protected final int yOutFrom
protected final int yOutTo
public CPUIm2ColTask(INDArray img, int kernelHeight, int kernelWidth, int strideY, int strideX, int padHeight, int padWidth, boolean coverAll, int parallelThreshold)
public CPUIm2ColTask(INDArray img, INDArray out, int kernelHeight, int kernelWidth, int strideY, int strideX, int padHeight, int padWidth, int exampleFrom, int exampleTo, int depthFrom, int depthTo, int yOutFrom, int yOutTo, int xOutFrom, int xOutTo, boolean coverAll, int parallelThreshold)
protected INDArray compute()
compute
in class RecursiveTask<INDArray>
public INDArray call()
Task
public INDArray invokeBlocking()
Task
invokeBlocking
in interface Task<INDArray>
public void invokeAsync()
Task
invokeAsync
in interface Task<INDArray>
public INDArray blockUntilComplete()
Task
blockUntilComplete
in interface Task<INDArray>
Copyright © 2016. All Rights Reserved.