Package ai.djl.mxnet.engine

package ai.djl.mxnet.engine

Contains implementations of interfaces within the DJL API for the Apache MXNet Engine.

See Also:

• MxEngine

Class	Description
CachedOp	The CachedOp is an internal helper that provides the core functionality to execute a MxSymbolBlock.
GradReq	An enum that indicates whether gradient is required.
MxDataType	Helper to convert between DataType an the MXNet internal DataTypes.
MxDeviceType	DeviceType is the MXNet equivalent of the types in Device.
MxEngine	The MxEngine is an implementation of the Engine based on the Apache MXNet Deep Learning Framework.
MxEngineProvider	MxEngineProvider is the MXNet implementation of EngineProvider.
MxGradientCollector	MxGradientCollector is the MXNet implementation of GradientCollector.
MxModel	MxModel is the MXNet implementation of Model.
MxNDArray	MxNDArray is the MXNet implementation of NDArray.
MxNDArrayIndexer	The NDArrayIndexer used by the MxNDArray.
MxNDManager	MxNDManager is the MXNet implementation of NDManager.
MxOpParams	An internal helper for creating the MXNet operator parameters.
MxParameterServer	MxParameterServer is the MXNet implementation of ParameterServer.
MxSymbolBlock	MxSymbolBlock is the MXNet implementation of SymbolBlock.
Symbol	Symbol is an internal helper for symbolic model graphs used by the SymbolBlock.