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 |
ComputationGraphConfiguration.cacheMode |
protected CacheMode |
MultiLayerConfiguration.cacheMode |
protected CacheMode |
MultiLayerConfiguration.Builder.cacheMode |
protected CacheMode |
NeuralNetConfiguration.cacheMode |
protected CacheMode |
NeuralNetConfiguration.Builder.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(@NonNull 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(@NonNull 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,
DataType dataType) |
abstract long |
MemoryReport.getMemoryBytes(MemoryType memoryType,
int minibatchSize,
MemoryUseMode memoryUseMode,
CacheMode cacheMode,
DataType dataType)
Get the memory estimate (in bytes) for the specified type of memory
|
long |
NetworkMemoryReport.getMemoryBytes(MemoryType memoryType,
int minibatchSize,
MemoryUseMode memoryUseMode,
CacheMode cacheMode,
DataType dataType) |
long |
MemoryReport.getTotalMemoryBytes(int minibatchSize,
@NonNull MemoryUseMode memoryUseMode,
@NonNull CacheMode cacheMode)
Get the total memory use in bytes for the given configuration (using the current ND4J data type)
|
long |
LayerMemoryReport.getTotalMemoryBytes(int minibatchSize,
@NonNull MemoryUseMode memoryUseMode,
@NonNull CacheMode cacheMode,
@NonNull DataType dataType) |
abstract long |
MemoryReport.getTotalMemoryBytes(int minibatchSize,
@NonNull MemoryUseMode memoryUseMode,
@NonNull CacheMode cacheMode,
@NonNull DataType dataType)
Get the total memory use in bytes for the given configuration
|
long |
NetworkMemoryReport.getTotalMemoryBytes(int minibatchSize,
@NonNull MemoryUseMode memoryUseMode,
@NonNull CacheMode cacheMode,
@NonNull DataType dataType) |
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(BaseRecurrentLayer layer,
NeuralNetConfiguration conf,
IActivation gateActivationFn,
INDArray input,
INDArray recurrentWeights,
INDArray originalInputWeights,
INDArray biases,
boolean training,
INDArray originalPrevOutputActivations,
INDArray originalPrevMemCellState,
boolean forBackprop,
boolean forwards,
String inputWeightKey,
INDArray maskArray,
boolean hasPeepholeConnections,
LSTMHelper helper,
CacheMode cacheMode,
LayerWorkspaceMgr workspaceMgr,
boolean isHelperAllowFallback)
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 © 2021. All rights reserved.