| Package | Description |
|---|---|
| org.deeplearning4j.nn.conf.memory |
| Modifier and Type | Method and Description |
|---|---|
static MemoryUseMode |
MemoryUseMode.valueOf(String name)
Returns the enum constant of this type with the specified name.
|
static MemoryUseMode[] |
MemoryUseMode.values()
Returns an array containing the constants of this enum type, in
the order they are declared.
|
| 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
|
Copyright © 2018. All rights reserved.