ai.djl.mxnet.engine

Class MxDataType

java.lang.Object

ai.djl.mxnet.engine.MxDataType

public final class MxDataType
extends java.lang.Object

Helper to convert between DataType an the MXNet internal DataTypes.

Method Summary

Modifier and Type	Method and Description
static ai.djl.ndarray.types.DataType	fromMx(java.lang.String mxType) Converts a MXNet type String into a DataType.
static java.lang.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 Detail

fromMx

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

Converts a MXNet type String into a DataType.

Parameters:

mxType - the type String to convert

Returns:

the DataType

toMx

public static java.lang.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