org.nd4j.linalg.api.parallel.tasks

Interface TaskFactory

All Known Implementing Classes:

CPUTaskFactory

public interface TaskFactory

A TaskFactory provides Task objects for each type of Op

Method Summary

Modifier and Type	Method and Description
Task<Double>	getAccumulationTask(Accumulation op)
Task<INDArray>	getAccumulationTask(Accumulation op, int... dimension)
Task<INDArray>	getCol2ImTask(INDArray col, int strideY, int strideX, int padHeight, int padWidth, int imgHeight, int imgWidth)
Task<INDArray>	getIm2ColTask(INDArray img, int kernelHeight, int kernelWidth, int strideY, int strideX, int padHeight, int padWidth, boolean coverAll)
Task<org.apache.commons.math3.util.Pair<Double,Integer>>	getIndexAccumulationTask(IndexAccumulation op)
Task<INDArray>	getIndexAccumulationTask(IndexAccumulation op, int... dimension)
Task<Void>	getScalarAction(ScalarOp op)
Task<Void>	getTransformAction(TransformOp op)
Task<Void>	getTransformAction(TransformOp op, int... dimension)
Task<Void>	getVectorOpAction(VectorOp op)

Method Detail

getTransformAction

Task<Void> getTransformAction(TransformOp op)

getTransformAction

Task<Void> getTransformAction(TransformOp op,
                              int... dimension)

getScalarAction

Task<Void> getScalarAction(ScalarOp op)

getAccumulationTask

Task<Double> getAccumulationTask(Accumulation op)

getAccumulationTask

Task<INDArray> getAccumulationTask(Accumulation op,
                                   int... dimension)

getIndexAccumulationTask

Task<org.apache.commons.math3.util.Pair<Double,Integer>> getIndexAccumulationTask(IndexAccumulation op)

getIndexAccumulationTask

Task<INDArray> getIndexAccumulationTask(IndexAccumulation op,
                                        int... dimension)

getVectorOpAction

Task<Void> getVectorOpAction(VectorOp op)

getIm2ColTask

Task<INDArray> getIm2ColTask(INDArray img,
                             int kernelHeight,
                             int kernelWidth,
                             int strideY,
                             int strideX,
                             int padHeight,
                             int padWidth,
                             boolean coverAll)

getCol2ImTask

Task<INDArray> getCol2ImTask(INDArray col,
                             int strideY,
                             int strideX,
                             int padHeight,
                             int padWidth,
                             int imgHeight,
                             int imgWidth)