Uses of Class
org.nd4j.linalg.jcublas.util.KernelParamsWrapper

Packages that use KernelParamsWrapper

Package	Description
org.nd4j.linalg.jcublas.util

Uses of KernelParamsWrapper in org.nd4j.linalg.jcublas.util

Methods in org.nd4j.linalg.jcublas.util that return KernelParamsWrapper

Modifier and Type	Method and Description
KernelParamsWrapper	KernelParamsWrapper.setResultArray(INDArray array) set the array that will contain the results, If the array is not set, then data from the device will not be copied to the host
KernelParamsWrapper	KernelParamsWrapper.setResultOp(Accumulation op, INDArray result) set the Op that this result is for