org.nd4j.linalg.cpu.nativecpu.workspace

Class CpuWorkspace

java.lang.Object

org.nd4j.linalg.memory.abstracts.Nd4jWorkspace

org.nd4j.linalg.cpu.nativecpu.workspace.CpuWorkspace

All Implemented Interfaces:

AutoCloseable, org.nd4j.linalg.api.memory.MemoryWorkspace

public class CpuWorkspace
extends Nd4jWorkspace

CPU-only MemoryWorkspace implementation

Author:

raver119@gmail.com

Nested Class Summary

Nested classes/interfaces inherited from class org.nd4j.linalg.memory.abstracts.Nd4jWorkspace

Nd4jWorkspace.GarbageWorkspaceReference

Field Summary

Fields inherited from class org.nd4j.linalg.memory.abstracts.Nd4jWorkspace

borrowingWorkspace, currentSize, cycleAllocations, cyclesCount, deviceId, deviceOffset, disabledCounter, externalAllocations, externalCount, guid, hostOffset, id, initialBlockSize, isBorrowed, isDebug, isInit, isLearning, isOpen, isOver, isUsed, lastCycleAllocations, maxCycle, memoryManager, pinnedAllocations, pinnedAllocationsSize, pinnedCount, previousWorkspace, resetPlanned, SAFETY_OFFSET, spilledAllocationsSize, stepsCount, stepsNumber, tagScope, threadId, trimmedMode, trimmedStep, workspace, workspaceConfiguration

Fields inherited from interface org.nd4j.linalg.api.memory.MemoryWorkspace

DEFAULT_ID

Constructor Summary

Constructors

Constructor and Description
CpuWorkspace(org.nd4j.linalg.api.memory.conf.WorkspaceConfiguration configuration)
CpuWorkspace(org.nd4j.linalg.api.memory.conf.WorkspaceConfiguration configuration, String workspaceId)
CpuWorkspace(org.nd4j.linalg.api.memory.conf.WorkspaceConfiguration configuration, String workspaceId, Integer deviceId)

Method Summary

Modifier and Type	Method and Description
protected void	clearExternalAllocations()
protected void	clearPinnedAllocations(boolean extended)
void	destroyWorkspace(boolean extended)
protected void	init()
protected void	resetWorkspace()

Methods inherited from class org.nd4j.linalg.memory.abstracts.Nd4jWorkspace

alloc, alloc, close, destroyWorkspace, enableDebug, free, getCurrentSize, getCyclesCount, getDeviceId, getDeviceOffset, getHostOffset, getId, getInitialBlockSize, getLastCycleAllocations, getMaxCycleAllocations, getNumberOfExternalAllocations, getNumberOfPinnedAllocations, getParentWorkspace, getPinnedSize, getSpilledSize, getStepNumber, getThisCycleAllocations, getThreadId, getWorkspaceConfiguration, initializeWorkspace, isScopeActive, notifyScopeBorrowed, notifyScopeEntered, notifyScopeLeft, reset, tagOutOfScopeUse, toggleWorkspaceUse, toString

Methods inherited from class java.lang.Object

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

Constructor Detail

CpuWorkspace

public CpuWorkspace(@NonNull
                    org.nd4j.linalg.api.memory.conf.WorkspaceConfiguration configuration)

CpuWorkspace

public CpuWorkspace(@NonNull
                    org.nd4j.linalg.api.memory.conf.WorkspaceConfiguration configuration,
                    @NonNull
                    String workspaceId)

CpuWorkspace

public CpuWorkspace(@NonNull
                    org.nd4j.linalg.api.memory.conf.WorkspaceConfiguration configuration,
                    @NonNull
                    String workspaceId,
                    Integer deviceId)

Method Detail

init

protected void init()

Overrides:

init in class Nd4jWorkspace

clearPinnedAllocations

protected void clearPinnedAllocations(boolean extended)

Specified by:

clearPinnedAllocations in class Nd4jWorkspace

clearExternalAllocations

protected void clearExternalAllocations()

Specified by:

clearExternalAllocations in class Nd4jWorkspace

destroyWorkspace

public void destroyWorkspace(boolean extended)

Specified by:

destroyWorkspace in interface org.nd4j.linalg.api.memory.MemoryWorkspace

Overrides:

destroyWorkspace in class Nd4jWorkspace

resetWorkspace

protected void resetWorkspace()

Specified by:

resetWorkspace in class Nd4jWorkspace