Uses of Class
org.nd4j.linalg.api.memory.enums.AllocationKind
-
Packages that use AllocationKind Package Description org.nd4j.linalg.api.memory org.nd4j.linalg.api.memory.enums -
-
Uses of AllocationKind in org.nd4j.linalg.api.memory
Methods in org.nd4j.linalg.api.memory with parameters of type AllocationKind Modifier and Type Method Description longAllocationsTracker. bytesOnDevice(AllocationKind kind, Integer deviceId)longDeviceAllocationsTracker. getState(@NonNull AllocationKind kind)voidAllocationsTracker. markAllocated(AllocationKind kind, Integer deviceId, long bytes)voidAllocationsTracker. markReleased(AllocationKind kind, Integer deviceId, long bytes)voidDeviceAllocationsTracker. updateState(@NonNull AllocationKind kind, long bytes) -
Uses of AllocationKind in org.nd4j.linalg.api.memory.enums
Methods in org.nd4j.linalg.api.memory.enums that return AllocationKind Modifier and Type Method Description static AllocationKindAllocationKind. 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.
-