org.nd4j.linalg.cpu.nativecpu

Class CpuTADManager

java.lang.Object

org.nd4j.linalg.cpu.nativecpu.CpuTADManager

All Implemented Interfaces:

TADManager

public class CpuTADManager
extends Object
implements TADManager

Author:

raver119@gmail.com

Constructor Summary

Constructors

Constructor and Description
CpuTADManager()

Method Summary

Modifier and Type	Method and Description
org.apache.commons.math3.util.Pair<DataBuffer,DataBuffer>	getTADOnlyShapeInfo(INDArray array, int[] dimension)
void	init(NativeOps nativeOps, ConstantHandler constantHandler)

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Detail

CpuTADManager

public CpuTADManager()

Method Detail

init

public void init(@NonNull
                 NativeOps nativeOps,
                 @NonNull
                 ConstantHandler constantHandler)

getTADOnlyShapeInfo

public org.apache.commons.math3.util.Pair<DataBuffer,DataBuffer> getTADOnlyShapeInfo(INDArray array,
                                                                                     int[] dimension)

Specified by:

getTADOnlyShapeInfo in interface TADManager