Package ai.djl.mxnet.engine

Class MxDataType

java.lang.Object

ai.djl.mxnet.engine.MxDataType

public final class MxDataType
extends Object

Helper to convert between DataType an the MXNet internal DataTypes.

Method Summary

Modifier and Type	Method	Description
static ai.djl.ndarray.types.DataType	fromMx(String mxType)	Converts a MXNet type String into a DataType.
static String	toMx(ai.djl.ndarray.types.DataType jType)	Converts a DataType into the corresponding MXNet type String.

Methods inherited from class java.lang.Object

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

Method Details

fromMx

public static ai.djl.ndarray.types.DataType fromMx(String mxType)

Converts a MXNet type String into a DataType.

Parameters:

mxType - the type String to convert

Returns:

the DataType

toMx

public static String toMx(ai.djl.ndarray.types.DataType jType)

Converts a DataType into the corresponding MXNet type String.

Parameters:

jType - the java DataType to convert

Returns:

the converted MXNet type string