public class AllocUtil
extends java.lang.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(java.lang.String allocMode)
Get the allocation mode from the context
|
static java.lang.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(java.lang.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(java.lang.String allocMode)
public static java.lang.String getAllocModeName(DataBuffer.AllocationMode allocationMode)
allocationMode
- public static DataBuffer.AllocationMode getAllocationModeFromContext()
public static void setAllocationModeForContext(DataBuffer.AllocationMode allocationModeForContext)
allocationModeForContext
- public static void setAllocationModeForContext(java.lang.String allocationModeForContext)
allocationModeForContext
- Copyright © 2019. All rights reserved.