public class DeviceTADManager extends BasicTADManager
Modifier and Type | Field and Description |
---|---|
protected List<Map<TadDescriptor,org.apache.commons.math3.util.Pair<DataBuffer,DataBuffer>>> |
tadCache |
nativeOps
Constructor and Description |
---|
DeviceTADManager() |
Modifier and Type | Method and Description |
---|---|
org.apache.commons.math3.util.Pair<DataBuffer,DataBuffer> |
getTADOnlyShapeInfo(INDArray array,
int[] dimension) |
void |
purgeBuffers()
This method removes all cached shape buffers
|
protected List<Map<TadDescriptor,org.apache.commons.math3.util.Pair<DataBuffer,DataBuffer>>> tadCache
public void purgeBuffers()
purgeBuffers
in interface TADManager
purgeBuffers
in class BasicTADManager
public org.apache.commons.math3.util.Pair<DataBuffer,DataBuffer> getTADOnlyShapeInfo(INDArray array, int[] dimension)
getTADOnlyShapeInfo
in interface TADManager
getTADOnlyShapeInfo
in class BasicTADManager
Copyright © 2016. All Rights Reserved.