| Package | Description |
|---|---|
| org.deeplearning4j.nn.conf.memory |
| Modifier and Type | Method and Description |
|---|---|
LayerMemoryReport.Builder |
LayerMemoryReport.Builder.cacheMemory(long cacheModeMemoryFixed,
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.standardMemory(long parameterSize,
long updaterStateSize)
Report the standard memory
|
LayerMemoryReport.Builder |
LayerMemoryReport.Builder.workingMemory(long fixedInference,
long variableInferencePerEx,
long fixedTrain,
long variableTrainPerEx)
Report the working memory size, 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.
|
| Constructor and Description |
|---|
LayerMemoryReport(LayerMemoryReport.Builder b) |
Copyright © 2018. All rights reserved.