Modifier and Type | Method and Description |
---|---|
void |
Layer.setCacheMode(CacheMode mode)
This method sets given CacheMode for current layer
|
Modifier and Type | Field and Description |
---|---|
protected CacheMode |
MultiLayerConfiguration.cacheMode |
protected CacheMode |
MultiLayerConfiguration.Builder.cacheMode |
protected CacheMode |
NeuralNetConfiguration.cacheMode |
protected CacheMode |
NeuralNetConfiguration.Builder.cacheMode |
protected CacheMode |
ComputationGraphConfiguration.cacheMode |
Modifier and Type | Method and Description |
---|---|
static CacheMode |
CacheMode.valueOf(String name)
Returns the enum constant of this type with the specified name.
|
static CacheMode[] |
CacheMode.values()
Returns an array containing the constants of this enum type, in
the order they are declared.
|
Modifier and Type | Method and Description |
---|---|
MultiLayerConfiguration.Builder |
MultiLayerConfiguration.Builder.cacheMode(CacheMode cacheMode)
This method defines how/if preOutput cache is handled:
NONE: cache disabled (default value)
HOST: Host memory will be used
DEVICE: GPU memory will be used (on CPU backends effect will be the same as for HOST)
|
NeuralNetConfiguration.Builder |
NeuralNetConfiguration.Builder.cacheMode(CacheMode cacheMode)
This method defines how/if preOutput cache is handled:
NONE: cache disabled (default value)
HOST: Host memory will be used
DEVICE: GPU memory will be used (on CPU backends effect will be the same as for HOST)
|
Modifier and Type | Field and Description |
---|---|
static Map<CacheMode,Long> |
MemoryReport.CACHE_MODE_ALL_ZEROS
A simple Map containing all zeros for each CacheMode key
|
Modifier and Type | Method and Description |
---|---|
static Map<CacheMode,Long> |
MemoryReport.cacheModeMapFor(long value)
Get a map of CacheMode with all keys associated with the specified value
|
Modifier and Type | Method and Description |
---|---|
long |
MemoryReport.getMemoryBytes(MemoryType memoryType,
int minibatchSize,
MemoryUseMode memoryUseMode,
CacheMode cacheMode)
Get the memory estimate (in bytes) for the specified type of memory, using the current ND4J data type
|
long |
LayerMemoryReport.getMemoryBytes(MemoryType memoryType,
int minibatchSize,
MemoryUseMode memoryUseMode,
CacheMode cacheMode,
org.nd4j.linalg.api.buffer.DataBuffer.Type dataType) |
long |
NetworkMemoryReport.getMemoryBytes(MemoryType memoryType,
int minibatchSize,
MemoryUseMode memoryUseMode,
CacheMode cacheMode,
org.nd4j.linalg.api.buffer.DataBuffer.Type dataType) |
abstract long |
MemoryReport.getMemoryBytes(MemoryType memoryType,
int minibatchSize,
MemoryUseMode memoryUseMode,
CacheMode cacheMode,
org.nd4j.linalg.api.buffer.DataBuffer.Type dataType)
Get the memory estimate (in bytes) for the specified type of memory
|
long |
MemoryReport.getTotalMemoryBytes(int minibatchSize,
MemoryUseMode memoryUseMode,
CacheMode cacheMode)
Get the total memory use in bytes for the given configuration (using the current ND4J data type)
|
long |
LayerMemoryReport.getTotalMemoryBytes(int minibatchSize,
MemoryUseMode memoryUseMode,
CacheMode cacheMode,
org.nd4j.linalg.api.buffer.DataBuffer.Type dataType) |
long |
NetworkMemoryReport.getTotalMemoryBytes(int minibatchSize,
MemoryUseMode memoryUseMode,
CacheMode cacheMode,
org.nd4j.linalg.api.buffer.DataBuffer.Type dataType) |
abstract long |
MemoryReport.getTotalMemoryBytes(int minibatchSize,
MemoryUseMode memoryUseMode,
CacheMode cacheMode,
org.nd4j.linalg.api.buffer.DataBuffer.Type dataType)
Get the total memory use in bytes for the given configuration
|
Modifier and Type | Method and Description |
---|---|
LayerMemoryReport.Builder |
LayerMemoryReport.Builder.cacheMemory(Map<CacheMode,Long> cacheModeMemoryFixed,
Map<CacheMode,Long> cacheModeMemoryVariablePerEx)
Reports the cached/cacheable memory requirements.
|
LayerMemoryReport.Builder |
LayerMemoryReport.Builder.cacheMemory(Map<CacheMode,Long> cacheModeMemoryFixed,
Map<CacheMode,Long> cacheModeMemoryVariablePerEx)
Reports the cached/cacheable memory requirements.
|
LayerMemoryReport.Builder |
LayerMemoryReport.Builder.workingMemory(long fixedInference,
long variableInferencePerEx,
Map<CacheMode,Long> fixedTrain,
Map<CacheMode,Long> variableTrainPerEx)
Report the working memory requirements, for both inference and training.
|
LayerMemoryReport.Builder |
LayerMemoryReport.Builder.workingMemory(long fixedInference,
long variableInferencePerEx,
Map<CacheMode,Long> fixedTrain,
Map<CacheMode,Long> variableTrainPerEx)
Report the working memory requirements, for both inference and training.
|
Modifier and Type | Method and Description |
---|---|
void |
ComputationGraph.setCacheMode(CacheMode mode)
This method sets specified CacheMode for all layers within network
|
Modifier and Type | Field and Description |
---|---|
protected CacheMode |
AbstractLayer.cacheMode |
Modifier and Type | Method and Description |
---|---|
void |
AbstractLayer.setCacheMode(CacheMode mode) |
void |
FrozenLayer.setCacheMode(CacheMode mode) |
Modifier and Type | Method and Description |
---|---|
static FwdPassReturn |
LSTMHelpers.activateHelper(BaseLayer layer,
NeuralNetConfiguration conf,
org.nd4j.linalg.activations.IActivation gateActivationFn,
org.nd4j.linalg.api.ndarray.INDArray input,
org.nd4j.linalg.api.ndarray.INDArray recurrentWeights,
org.nd4j.linalg.api.ndarray.INDArray originalInputWeights,
org.nd4j.linalg.api.ndarray.INDArray biases,
boolean training,
org.nd4j.linalg.api.ndarray.INDArray originalPrevOutputActivations,
org.nd4j.linalg.api.ndarray.INDArray originalPrevMemCellState,
boolean forBackprop,
boolean forwards,
String inputWeightKey,
org.nd4j.linalg.api.ndarray.INDArray maskArray,
boolean hasPeepholeConnections,
LSTMHelper helper,
CacheMode cacheMode,
LayerWorkspaceMgr workspaceMgr)
Returns FwdPassReturn object with activations/INDArrays.
|
void |
BidirectionalLayer.setCacheMode(CacheMode mode) |
Modifier and Type | Field and Description |
---|---|
protected CacheMode |
VariationalAutoencoder.cacheMode |
Modifier and Type | Method and Description |
---|---|
void |
VariationalAutoencoder.setCacheMode(CacheMode mode) |
Modifier and Type | Method and Description |
---|---|
void |
BaseWrapperLayer.setCacheMode(CacheMode mode) |
Modifier and Type | Method and Description |
---|---|
void |
MultiLayerNetwork.setCacheMode(CacheMode mode)
This method sets specified CacheMode for all layers within network
|
Copyright © 2018. All rights reserved.