Interface TADManager


  • public interface TADManager
    • 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: