org.deeplearning4j.nn.workspace

Class LayerWorkspaceMgr

java.lang.Object

org.nd4j.linalg.workspace.BaseWorkspaceMgr<ArrayType>

org.deeplearning4j.nn.workspace.LayerWorkspaceMgr

All Implemented Interfaces:

org.nd4j.linalg.workspace.WorkspaceMgr<ArrayType>

public class LayerWorkspaceMgr
extends org.nd4j.linalg.workspace.BaseWorkspaceMgr<ArrayType>

WorkspaceMgr for DL4J layers. Used to flexibly specify which workspaces a given array type (defined by ArrayType) should be placed in

Author:

Alex Black

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
static class	LayerWorkspaceMgr.Builder

Field Summary

Fields

Modifier and Type	Field and Description
protected Set<String>	noLeverageOverride

Fields inherited from class org.nd4j.linalg.workspace.BaseWorkspaceMgr

configMap, scopeOutOfWs, workspaceNames

Constructor Summary

Constructors

Constructor and Description
LayerWorkspaceMgr(Set<ArrayType> scopeOutOfWs, Map<ArrayType,org.nd4j.linalg.api.memory.conf.WorkspaceConfiguration> configMap, Map<ArrayType,String> workspaceNames)

Method Summary

Modifier and Type	Method and Description
static LayerWorkspaceMgr.Builder	builder()
org.nd4j.linalg.api.ndarray.INDArray	leverageTo(ArrayType arrayType, org.nd4j.linalg.api.ndarray.INDArray array)
static LayerWorkspaceMgr	noWorkspaces()
static LayerWorkspaceMgr	noWorkspacesImmutable()
void	setNoLeverageOverride(String wsName)
org.nd4j.linalg.api.ndarray.INDArray	validateArrayLocation(ArrayType arrayType, org.nd4j.linalg.api.ndarray.INDArray array, boolean migrateIfInvalid, boolean exceptionIfDetached)

Methods inherited from class org.nd4j.linalg.workspace.BaseWorkspaceMgr

assertCurrentWorkspace, assertNotOpen, assertOpen, create, create, createUninitialized, createUninitialized, dup, dup, getConfiguration, getWorkspaceName, hasConfiguration, isScopedOut, isWorkspaceOpen, notifyScopeBorrowed, notifyScopeEntered, notifyScopeEntered, setConfiguration, setScopedOutFor, setWorkspace, setWorkspaceName

Methods inherited from class java.lang.Object

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

Field Detail

noLeverageOverride

protected Set<String> noLeverageOverride

Constructor Detail

LayerWorkspaceMgr

public LayerWorkspaceMgr(Set<ArrayType> scopeOutOfWs,
                         Map<ArrayType,org.nd4j.linalg.api.memory.conf.WorkspaceConfiguration> configMap,
                         Map<ArrayType,String> workspaceNames)

Method Detail

setNoLeverageOverride

public void setNoLeverageOverride(String wsName)

leverageTo

public org.nd4j.linalg.api.ndarray.INDArray leverageTo(ArrayType arrayType,
                                                       org.nd4j.linalg.api.ndarray.INDArray array)

Specified by:

leverageTo in interface org.nd4j.linalg.workspace.WorkspaceMgr<ArrayType>

Overrides:

leverageTo in class org.nd4j.linalg.workspace.BaseWorkspaceMgr<ArrayType>

validateArrayLocation

public org.nd4j.linalg.api.ndarray.INDArray validateArrayLocation(@NonNull
                                                                  ArrayType arrayType,
                                                                  @NonNull
                                                                  org.nd4j.linalg.api.ndarray.INDArray array,
                                                                  boolean migrateIfInvalid,
                                                                  boolean exceptionIfDetached)

Specified by:

validateArrayLocation in interface org.nd4j.linalg.workspace.WorkspaceMgr<ArrayType>

Overrides:

validateArrayLocation in class org.nd4j.linalg.workspace.BaseWorkspaceMgr<ArrayType>

builder

public static LayerWorkspaceMgr.Builder builder()

noWorkspaces

public static LayerWorkspaceMgr noWorkspaces()

noWorkspacesImmutable

public static LayerWorkspaceMgr noWorkspacesImmutable()