Package | Description |
---|---|
org.nd4j.linalg.api.memory | |
org.nd4j.linalg.api.memory.enums |
Modifier and Type | Method and Description |
---|---|
long |
AllocationsTracker.bytesOnDevice(AllocationKind kind,
java.lang.Integer deviceId) |
long |
DeviceAllocationsTracker.getState(AllocationKind kind) |
void |
AllocationsTracker.markAllocated(AllocationKind kind,
java.lang.Integer deviceId,
long bytes) |
void |
AllocationsTracker.markReleased(AllocationKind kind,
java.lang.Integer deviceId,
long bytes) |
void |
DeviceAllocationsTracker.updateState(AllocationKind kind,
long bytes) |
Modifier and Type | Method and Description |
---|---|
static AllocationKind |
AllocationKind.valueOf(java.lang.String name)
Returns the enum constant of this type with the specified name.
|
static AllocationKind[] |
AllocationKind.values()
Returns an array containing the constants of this enum type, in
the order they are declared.
|
Copyright © 2019. All rights reserved.