| Package | Description |
|---|---|
| org.nd4j.linalg.api.memory | |
| org.nd4j.linalg.api.memory.enums |
| Modifier and Type | Method and Description |
|---|---|
long |
AllocationsTracker.bytesOnDevice(AllocationKind kind,
Integer deviceId) |
long |
DeviceAllocationsTracker.getState(@NonNull AllocationKind kind) |
void |
AllocationsTracker.markAllocated(AllocationKind kind,
Integer deviceId,
long bytes) |
void |
AllocationsTracker.markReleased(AllocationKind kind,
Integer deviceId,
long bytes) |
void |
DeviceAllocationsTracker.updateState(@NonNull AllocationKind kind,
long bytes) |
| Modifier and Type | Method and Description |
|---|---|
static AllocationKind |
AllocationKind.valueOf(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 © 2021. All rights reserved.