org.deeplearning4j.nn.conf.memory

Class LayerMemoryReport

java.lang.Object

org.deeplearning4j.nn.conf.memory.MemoryReport

org.deeplearning4j.nn.conf.memory.LayerMemoryReport

public class LayerMemoryReport
extends MemoryReport

A MemoryReport Designed to report estimated memory use for a single layer or graph vertex.

Author:

Alex Black

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
static class	LayerMemoryReport.Builder

Field Summary

Fields inherited from class org.deeplearning4j.nn.conf.memory.MemoryReport

CACHE_MODE_ALL_ZEROS

Constructor Summary

Constructors

Modifier	Constructor and Description
protected	LayerMemoryReport(LayerMemoryReport.Builder b)

Method Summary

Modifier and Type	Method and Description
long	getMemoryBytes(MemoryType memoryType, int minibatchSize, MemoryUseMode memoryUseMode, CacheMode cacheMode, org.nd4j.linalg.api.buffer.DataType dataType) Get the memory estimate (in bytes) for the specified type of memory
String	getName() Name of the object that the memory report was generated for
Class<?>	getReportClass()
long	getTotalMemoryBytes(int minibatchSize, MemoryUseMode memoryUseMode, CacheMode cacheMode, org.nd4j.linalg.api.buffer.DataType dataType) Get the total memory use in bytes for the given configuration
void	scale(int scale) Multiply all memory usage by the specified scaling factor
String	toString()

Methods inherited from class org.deeplearning4j.nn.conf.memory.MemoryReport

cacheModeMapFor, fromJson, fromYaml, getBytesPerElement, getMemoryBytes, getTotalMemoryBytes, toJson, toYaml

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, wait, wait, wait

Constructor Detail

LayerMemoryReport

protected LayerMemoryReport(LayerMemoryReport.Builder b)

Method Detail

getReportClass

public Class<?> getReportClass()

Specified by:

getReportClass in class MemoryReport

Returns:

Class that the memory report was generated for

getName

public String getName()

Description copied from class: MemoryReport

Name of the object that the memory report was generated for

Specified by:

getName in class MemoryReport

Returns:

Name of the object

getTotalMemoryBytes

public long getTotalMemoryBytes(int minibatchSize,
                                @NonNull
                                MemoryUseMode memoryUseMode,
                                @NonNull
                                CacheMode cacheMode,
                                @NonNull
                                org.nd4j.linalg.api.buffer.DataType dataType)

Description copied from class: MemoryReport

Get the total memory use in bytes for the given configuration

Specified by:

getTotalMemoryBytes in class MemoryReport

Parameters:

minibatchSize - Mini batch size to estimate the memory for

memoryUseMode - The memory use mode (training or inference)

cacheMode - The CacheMode to use

dataType - Nd4j datatype

Returns:

The estimated total memory consumption in bytes

getMemoryBytes

public long getMemoryBytes(MemoryType memoryType,
                           int minibatchSize,
                           MemoryUseMode memoryUseMode,
                           CacheMode cacheMode,
                           org.nd4j.linalg.api.buffer.DataType dataType)

Description copied from class: MemoryReport

Get the memory estimate (in bytes) for the specified type of memory

Specified by:

getMemoryBytes in class MemoryReport

Parameters:

memoryType - Type of memory to get the estimate for invites

minibatchSize - Mini batch size to estimate the memory for

memoryUseMode - The memory use mode (training or inference)

cacheMode - The CacheMode to use

dataType - Nd4j datatype

Returns:

Estimated memory use for the given memory type

toString

public String toString()

Specified by:

toString in class MemoryReport

scale

public void scale(int scale)

Multiply all memory usage by the specified scaling factor

Parameters:

scale - Scale factor to multiply all memory usage by