public abstract class BaseShapeInfoProvider extends Object implements ShapeInfoProvider
Modifier and Type | Field and Description |
---|---|
protected AtomicLong |
bytes |
Constructor and Description |
---|
BaseShapeInfoProvider() |
Modifier and Type | Method and Description |
---|---|
org.nd4j.linalg.primitives.Pair<DataBuffer,long[]> |
createShapeInformation(long[] shape,
char order,
DataType dataType)
This method creates shapeInformation buffer, based on shape & order being passed in
|
org.nd4j.linalg.primitives.Pair<DataBuffer,long[]> |
createShapeInformation(long[] shape,
DataType dataType)
This method creates shapeInformation buffer, based on shape being passed in
|
org.nd4j.linalg.primitives.Pair<DataBuffer,long[]> |
createShapeInformation(long[] shape,
long[] stride,
long elementWiseStride,
char order,
DataType dataType)
This method creates long shapeInformation buffer, based on detailed shape info being passed in
|
org.nd4j.linalg.primitives.Pair<DataBuffer,long[]> |
createShapeInformation(long[] shape,
long[] stride,
long elementWiseStride,
char order,
long extras) |
long |
getCachedBytes()
This method returns memory used for cache, in bytes
|
clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait
purgeCache
protected AtomicLong bytes
public org.nd4j.linalg.primitives.Pair<DataBuffer,long[]> createShapeInformation(long[] shape, DataType dataType)
createShapeInformation
in interface ShapeInfoProvider
shape
- public org.nd4j.linalg.primitives.Pair<DataBuffer,long[]> createShapeInformation(long[] shape, char order, DataType dataType)
createShapeInformation
in interface ShapeInfoProvider
shape
- order
- public org.nd4j.linalg.primitives.Pair<DataBuffer,long[]> createShapeInformation(long[] shape, long[] stride, long elementWiseStride, char order, DataType dataType)
ShapeInfoProvider
createShapeInformation
in interface ShapeInfoProvider
public org.nd4j.linalg.primitives.Pair<DataBuffer,long[]> createShapeInformation(long[] shape, long[] stride, long elementWiseStride, char order, long extras)
createShapeInformation
in interface ShapeInfoProvider
public long getCachedBytes()
ShapeInfoProvider
getCachedBytes
in interface ShapeInfoProvider
Copyright © 2019. All rights reserved.