/** * 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 * @param allocationModeForContext */ public static void setDTypeForContext(DataBuffer.Type allocationModeForContext) { try { lock.writeLock().lock(); dtype = allocationModeForContext; setDTypeForContext(getDTypeForName(allocationModeForContext)); } finally { lock.writeLock().unlock(); } }