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 and Description
Builder()

Method Summary

Modifier and Type	Method and Description
LayerWorkspaceMgr	build()
LayerWorkspaceMgr.Builder	defaultNoWorkspace() Set the default to be scoped out for all array types.
LayerWorkspaceMgr.Builder	defaultWorkspace(String workspaceName, org.nd4j.linalg.api.memory.conf.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, org.nd4j.linalg.api.memory.conf.WorkspaceConfiguration configuration) Configure the workspace (name, configuration) for the specified array type

Methods inherited from class java.lang.Object

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

Constructor Detail

Builder

public Builder()

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,
                                                  org.nd4j.linalg.api.memory.conf.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 types

configuration - Configuration to use for all (not set) arrray types

Returns:

Builder

with

public LayerWorkspaceMgr.Builder with(ArrayType type,
                                      String workspaceName,
                                      org.nd4j.linalg.api.memory.conf.WorkspaceConfiguration configuration)

Configure the workspace (name, configuration) for the specified array type

Parameters:

type - Array type

workspaceName - Workspace name for the specified array type

configuration - Configuration for the specified array type

Returns:

Builder

build

public LayerWorkspaceMgr build()