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
long	getCachedBytes()
org.nd4j.linalg.primitives.Pair<org.nd4j.linalg.api.buffer.DataBuffer,org.nd4j.linalg.api.buffer.DataBuffer>	getTADOnlyShapeInfo(INDArray array, int[] dimension)
void	init(NativeOps nativeOps, ConstantHandler constantHandler)
void	purgeBuffers() This method removes all cached shape buffers

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)

purgeBuffers

public void purgeBuffers()

This method removes all cached shape buffers

Specified by:

purgeBuffers in interface TADManager

getTADOnlyShapeInfo

public org.nd4j.linalg.primitives.Pair<org.nd4j.linalg.api.buffer.DataBuffer,org.nd4j.linalg.api.buffer.DataBuffer> getTADOnlyShapeInfo(INDArray array,
                                                                                                                                        int[] dimension)

Specified by:

getTADOnlyShapeInfo in interface TADManager

getCachedBytes

public long getCachedBytes()

Specified by:

getCachedBytes in interface TADManager