public class AllocUtil extends Object
Constructor and Description |
---|
AllocUtil() |
Modifier and Type | Method and Description |
---|---|
static DataBuffer.AllocationMode |
getAllocationModeFromContext()
get the allocation mode from the context
|
static DataBuffer.AllocationMode |
getAllocationModeFromContext(String allocMode)
Get the allocation mode from the context
|
static String |
getAllocModeName(DataBuffer.AllocationMode allocationMode)
Gets the name of the alocation mode
|
static void |
setAllocationModeForContext(DataBuffer.AllocationMode allocationModeForContext)
Set the allocation mode for the nd4j context
The value must be one of: heap, java cpp, or direct
or an @link{IllegalArgumentException} is thrown
|
static void |
setAllocationModeForContext(String allocationModeForContext)
Set the allocation mode for the nd4j context
The value must be one of: heap, java cpp, or direct
or an @link{IllegalArgumentException} is thrown
|
public static DataBuffer.AllocationMode getAllocationModeFromContext(String allocMode)
public static String getAllocModeName(DataBuffer.AllocationMode allocationMode)
allocationMode
- public static DataBuffer.AllocationMode getAllocationModeFromContext()
public static void setAllocationModeForContext(DataBuffer.AllocationMode allocationModeForContext)
allocationModeForContext
- public static void setAllocationModeForContext(String allocationModeForContext)
allocationModeForContext
- Copyright © 2018. All rights reserved.