Package org.nd4j.linalg.cache
Interface TADManager
-
public interface TADManager
-
-
Method Summary
All Methods Instance Methods Abstract Methods Modifier and Type Method Description longgetCachedBytes()This method returns memory used for cache, in bytesPair<DataBuffer,DataBuffer>getTADOnlyShapeInfo(INDArray array, int... dimension)This method returns TAD shapeInfo and all offsets for specified tensor and dimensions.voidpurgeBuffers()This method removes all cached shape buffers
-
-
-
Method Detail
-
getTADOnlyShapeInfo
Pair<DataBuffer,DataBuffer> getTADOnlyShapeInfo(INDArray array, int... dimension)
This method returns TAD shapeInfo and all offsets for specified tensor and dimensions.- Parameters:
array- Tensor for TAD precalculationdimension-- Returns:
-
purgeBuffers
void purgeBuffers()
This method removes all cached shape buffers
-
getCachedBytes
long getCachedBytes()
This method returns memory used for cache, in bytes- Returns:
-
-