org.deeplearning4j.nn.layers.samediff

Class DL4JSameDiffMemoryMgr

java.lang.Object

org.nd4j.autodiff.samediff.internal.memory.AbstractMemoryMgr

org.deeplearning4j.nn.layers.samediff.DL4JSameDiffMemoryMgr

All Implemented Interfaces:

Closeable, AutoCloseable, SessionMemMgr

public class DL4JSameDiffMemoryMgr
extends AbstractMemoryMgr

A SameDiff SessionMemMgr that uses DL4J workspaces for memory management. Any op outputs are allocated in the output workspace if they are returned to the layer; otherwise they are placed in the DL4J working memory workspace

Author:

Alex Black

Constructor Summary

Constructors

Constructor and Description
DL4JSameDiffMemoryMgr(String workingMemoryWs, String outputWs, org.nd4j.linalg.api.memory.conf.WorkspaceConfiguration confWorking, org.nd4j.linalg.api.memory.conf.WorkspaceConfiguration confOutput)

Method Summary

Modifier and Type	Method and Description
INDArray	allocate(boolean detached, org.nd4j.linalg.api.buffer.DataType dataType, long... shape)
INDArray	allocate(boolean detached, LongShapeDescriptor descriptor)
void	close()
void	release(INDArray array)

Methods inherited from class org.nd4j.autodiff.samediff.internal.memory.AbstractMemoryMgr

dup, ulike

Methods inherited from class java.lang.Object

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

Constructor Detail

DL4JSameDiffMemoryMgr

public DL4JSameDiffMemoryMgr(String workingMemoryWs,
                             String outputWs,
                             org.nd4j.linalg.api.memory.conf.WorkspaceConfiguration confWorking,
                             org.nd4j.linalg.api.memory.conf.WorkspaceConfiguration confOutput)

Method Detail

allocate

public INDArray allocate(boolean detached,
                         org.nd4j.linalg.api.buffer.DataType dataType,
                         long... shape)

allocate

public INDArray allocate(boolean detached,
                         LongShapeDescriptor descriptor)

release

public void release(INDArray array)

close

public void close()