org.nd4j.linalg.api.buffer.util

Class AllocUtil

java.lang.Object

org.nd4j.linalg.api.buffer.util.AllocUtil

public class AllocUtil
extends Object

Used for manipulating the allocation variable in nd4j's context

Author:

Adam Gibson

Constructor Summary

Constructors

Constructor and Description
AllocUtil()

Method Summary

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

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Detail

AllocUtil

public AllocUtil()

Method Detail

getAllocationModeFromContext

public static DataBuffer.AllocationMode getAllocationModeFromContext(String allocMode)

Get the allocation mode from the context

Returns:

getAllocModeName

public static String getAllocModeName(DataBuffer.AllocationMode allocationMode)

Gets the name of the alocation mode

Parameters:

allocationMode -

Returns:

getAllocationModeFromContext

public static DataBuffer.AllocationMode getAllocationModeFromContext()

get the allocation mode from the context

Returns:

setAllocationModeForContext

public 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

Parameters:

allocationModeForContext -

setAllocationModeForContext

public 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

Parameters:

allocationModeForContext -