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,
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) |
Copyright © 2020. All rights reserved.