org.nd4j.linalg.cache

Interface TADManager

public interface TADManager

This interface describes TAD caching. While working with tensors, all operations are happening on some demensions, and since training tasks are repetitive, we can pre-calculate TAD shapes and offsets once, and reuse them later during whole training process.

Author:

[email protected]

Method Summary

Modifier and Type	Method and Description
long	getCachedBytes() This method returns memory used for cache, in bytes
Pair<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 precalculation

dimension -

Returns:

purgeBuffers

void purgeBuffers()

This method removes all cached shape buffers

getCachedBytes

long getCachedBytes()

This method returns memory used for cache, in bytes

Returns: