public class CpuTADManager extends Object implements TADManager
| Constructor and Description |
|---|
CpuTADManager() |
| 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
|
public void init(@NonNull
NativeOps nativeOps,
@NonNull
ConstantHandler constantHandler)
public void purgeBuffers()
purgeBuffers in interface TADManagerpublic org.nd4j.linalg.primitives.Pair<org.nd4j.linalg.api.buffer.DataBuffer,org.nd4j.linalg.api.buffer.DataBuffer> getTADOnlyShapeInfo(INDArray array, int[] dimension)
getTADOnlyShapeInfo in interface TADManagerpublic long getCachedBytes()
getCachedBytes in interface TADManagerCopyright © 2017. All rights reserved.