ai.djl.mxnet.engine

Class MxEngine

java.lang.Object

ai.djl.engine.Engine

ai.djl.mxnet.engine.MxEngine

public class MxEngine
extends ai.djl.engine.Engine

The MxEngine is an implementation of the Engine based on the Apache MXNet Deep Learning Framework.

To get an instance of the MxEngine when it is not the default Engine, call Engine.getEngine(String) with the Engine name "MXNet".

Field Summary

Fields

Modifier and Type	Field and Description
static java.lang.String	ENGINE_NAME

Method Summary

Modifier and Type	Method and Description
ai.djl.Device	defaultDevice()
java.lang.String	getEngineName()
int	getGpuCount()
java.lang.management.MemoryUsage	getGpuMemory(ai.djl.Device device)
java.lang.String	getVersion()
ai.djl.ndarray.NDManager	newBaseManager()
ai.djl.ndarray.NDManager	newBaseManager(ai.djl.Device device)
ai.djl.Model	newModel(ai.djl.Device device)

Methods inherited from class ai.djl.engine.Engine

getEngine, getInstance

Methods inherited from class java.lang.Object

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

Field Detail

ENGINE_NAME

public static final java.lang.String ENGINE_NAME

See Also:

Constant Field Values

Method Detail

getEngineName

public java.lang.String getEngineName()

Specified by:

getEngineName in class ai.djl.engine.Engine

getGpuCount

public int getGpuCount()

Specified by:

getGpuCount in class ai.djl.engine.Engine

getGpuMemory

public java.lang.management.MemoryUsage getGpuMemory(ai.djl.Device device)

Specified by:

getGpuMemory in class ai.djl.engine.Engine

defaultDevice

public ai.djl.Device defaultDevice()

Specified by:

defaultDevice in class ai.djl.engine.Engine

getVersion

public java.lang.String getVersion()

Specified by:

getVersion in class ai.djl.engine.Engine

newModel

public ai.djl.Model newModel(ai.djl.Device device)

Specified by:

newModel in class ai.djl.engine.Engine

newBaseManager

public ai.djl.ndarray.NDManager newBaseManager()

Specified by:

newBaseManager in class ai.djl.engine.Engine

newBaseManager

public ai.djl.ndarray.NDManager newBaseManager(ai.djl.Device device)

Specified by:

newBaseManager in class ai.djl.engine.Engine