Package org.deeplearning4j.nn.workspace
Class LayerWorkspaceMgr.Builder
- java.lang.Object
-
- org.deeplearning4j.nn.workspace.LayerWorkspaceMgr.Builder
-
- Enclosing class:
- LayerWorkspaceMgr
public static class LayerWorkspaceMgr.Builder extends Object
-
-
Constructor Summary
Constructors Constructor Description Builder()
-
Method Summary
All Methods Instance Methods Concrete Methods Modifier and Type Method Description LayerWorkspaceMgr
build()
LayerWorkspaceMgr.Builder
defaultNoWorkspace()
Set the default to be scoped out for all array types.LayerWorkspaceMgr.Builder
defaultWorkspace(String workspaceName, WorkspaceConfiguration configuration)
Set the default workspace for all array types to the specified workspace name/configuration NOTE: This will NOT override any settings previously set.LayerWorkspaceMgr.Builder
noWorkspaceFor(ArrayType type)
Specify that no workspace should be used for array of the specified type - i.e., these arrays should all be scoped out.LayerWorkspaceMgr.Builder
with(ArrayType type, String workspaceName, WorkspaceConfiguration configuration)
Configure the workspace (name, configuration) for the specified array type
-
-
-
Method Detail
-
defaultNoWorkspace
public LayerWorkspaceMgr.Builder defaultNoWorkspace()
Set the default to be scoped out for all array types. NOTE: Will not override the configuration for any array types that have already been configured- Returns:
- Builder
-
noWorkspaceFor
public LayerWorkspaceMgr.Builder noWorkspaceFor(ArrayType type)
Specify that no workspace should be used for array of the specified type - i.e., these arrays should all be scoped out.- Parameters:
type
- Array type to set scoped out for- Returns:
- Builder
-
defaultWorkspace
public LayerWorkspaceMgr.Builder defaultWorkspace(String workspaceName, WorkspaceConfiguration configuration)
Set the default workspace for all array types to the specified workspace name/configuration NOTE: This will NOT override any settings previously set.- Parameters:
workspaceName
- Name of the workspace to use for all (not set) arrray typesconfiguration
- Configuration to use for all (not set) arrray types- Returns:
- Builder
-
with
public LayerWorkspaceMgr.Builder with(ArrayType type, String workspaceName, WorkspaceConfiguration configuration)
Configure the workspace (name, configuration) for the specified array type- Parameters:
type
- Array typeworkspaceName
- Workspace name for the specified array typeconfiguration
- Configuration for the specified array type- Returns:
- Builder
-
build
public LayerWorkspaceMgr build()
-
-