public class CpuWorkspaceManager extends BasicWorkspaceManager
BasicWorkspaceManager.WorkspaceDeallocatorThreadbackingMap, defaultConfiguration| Constructor and Description |
|---|
CpuWorkspaceManager() |
| Modifier and Type | Method and Description |
|---|---|
org.nd4j.linalg.api.memory.MemoryWorkspace |
createNewWorkspace() |
org.nd4j.linalg.api.memory.MemoryWorkspace |
createNewWorkspace(org.nd4j.linalg.api.memory.conf.WorkspaceConfiguration configuration) |
org.nd4j.linalg.api.memory.MemoryWorkspace |
createNewWorkspace(org.nd4j.linalg.api.memory.conf.WorkspaceConfiguration configuration,
String id) |
org.nd4j.linalg.api.memory.MemoryWorkspace |
createNewWorkspace(org.nd4j.linalg.api.memory.conf.WorkspaceConfiguration configuration,
String id,
Integer deviceId) |
org.nd4j.linalg.api.memory.MemoryWorkspace |
getWorkspaceForCurrentThread(org.nd4j.linalg.api.memory.conf.WorkspaceConfiguration configuration,
String id) |
checkIfWorkspaceExists, destroyAllWorkspacesForCurrentThread, destroyWorkspace, destroyWorkspace, ensureThreadExistense, getAndActivateWorkspace, getAndActivateWorkspace, getAndActivateWorkspace, getWorkspaceForCurrentThread, getWorkspaceForCurrentThread, pickReference, printAllocationStatisticsForCurrentThread, scopeOutOfWorkspaces, setDefaultWorkspaceConfiguration, setWorkspaceForCurrentThread, setWorkspaceForCurrentThreadpublic org.nd4j.linalg.api.memory.MemoryWorkspace createNewWorkspace(@NonNull
org.nd4j.linalg.api.memory.conf.WorkspaceConfiguration configuration)
public org.nd4j.linalg.api.memory.MemoryWorkspace createNewWorkspace()
public org.nd4j.linalg.api.memory.MemoryWorkspace createNewWorkspace(@NonNull
org.nd4j.linalg.api.memory.conf.WorkspaceConfiguration configuration,
@NonNull
String id)
public org.nd4j.linalg.api.memory.MemoryWorkspace createNewWorkspace(@NonNull
org.nd4j.linalg.api.memory.conf.WorkspaceConfiguration configuration,
@NonNull
String id,
Integer deviceId)
public org.nd4j.linalg.api.memory.MemoryWorkspace getWorkspaceForCurrentThread(@NonNull
org.nd4j.linalg.api.memory.conf.WorkspaceConfiguration configuration,
@NonNull
String id)
Copyright © 2017. All rights reserved.