public class ConstantBuffersCache extends BasicConstantHandler
| Modifier and Type | Field and Description |
|---|---|
protected Map<ArrayDescriptor,org.nd4j.linalg.api.buffer.DataBuffer> |
buffersCache |
| Constructor and Description |
|---|
ConstantBuffersCache() |
| Modifier and Type | Method and Description |
|---|---|
long |
getCachedBytes() |
org.nd4j.linalg.api.buffer.DataBuffer |
getConstantBuffer(double[] array) |
org.nd4j.linalg.api.buffer.DataBuffer |
getConstantBuffer(float[] array) |
org.nd4j.linalg.api.buffer.DataBuffer |
getConstantBuffer(int[] array) |
void |
purgeConstants()
This method removes all cached constants
|
moveToConstantSpace, relocateConstantSpaceprotected Map<ArrayDescriptor,org.nd4j.linalg.api.buffer.DataBuffer> buffersCache
public org.nd4j.linalg.api.buffer.DataBuffer getConstantBuffer(int[] array)
public void purgeConstants()
public org.nd4j.linalg.api.buffer.DataBuffer getConstantBuffer(float[] array)
public org.nd4j.linalg.api.buffer.DataBuffer getConstantBuffer(double[] array)
public long getCachedBytes()
Copyright © 2017. All rights reserved.