Package org.nd4j.linalg.cache
Interface TADManager
-
public interface TADManager
-
-
Method Summary
All Methods Instance Methods Abstract Methods Modifier and Type Method Description long
getCachedBytes()
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.void
purgeBuffers()
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:
-
-