Package | Description |
---|---|
org.deeplearning4j.nn.workspace |
Modifier and Type | Method and Description |
---|---|
static LayerWorkspaceMgr.Builder |
LayerWorkspaceMgr.builder() |
LayerWorkspaceMgr.Builder |
LayerWorkspaceMgr.Builder.defaultNoWorkspace()
Set the default to be scoped out for all array types.
|
LayerWorkspaceMgr.Builder |
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 |
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 |
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
|
Copyright © 2018. All rights reserved.