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
-
-
Constructor Summary
Constructors Constructor Description DL4JSameDiffMemoryMgr(String workingMemoryWs, String outputWs, WorkspaceConfiguration confWorking, WorkspaceConfiguration confOutput)
-
Method Summary
All Methods Instance Methods Concrete Methods Modifier and Type Method Description INDArray
allocate(boolean detached, 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
-
-
-
-
Constructor Detail
-
DL4JSameDiffMemoryMgr
public DL4JSameDiffMemoryMgr(String workingMemoryWs, String outputWs, WorkspaceConfiguration confWorking, WorkspaceConfiguration confOutput)
-
-
Method Detail
-
allocate
public INDArray allocate(boolean detached, LongShapeDescriptor descriptor)
-
release
public void release(INDArray array)
-
close
public void close()
-
-