Package ai.djl.onnxruntime.engine
Class OrtEngine
java.lang.Object
ai.djl.engine.Engine
ai.djl.onnxruntime.engine.OrtEngine
public final class OrtEngine
extends ai.djl.engine.Engine
The
OrtEngine
is an implementation of the Engine
based on the ONNX Runtime Deep Learning Library.
To get an instance of the OrtEngine
when it is not the default Engine, call Engine.getEngine(String)
with the Engine name "OnnxRuntime".
-
Field Summary
Fields -
Method Summary
Modifier and TypeMethodDescriptionai.djl.engine.Engine
int
getRank()
boolean
hasCapability
(String capability) ai.djl.ndarray.NDManager
ai.djl.ndarray.NDManager
newBaseManager
(ai.djl.Device device) ai.djl.Model
toString()
Methods inherited from class ai.djl.engine.Engine
debugEnvironment, defaultDevice, getAllEngines, getDefaultEngineName, getDevices, getDevices, getDjlVersion, getEngine, getGpuCount, getInstance, getSeed, hasEngine, newGradientCollector, newParameterServer, newSymbolBlock, registerEngine, setRandomSeed
-
Field Details
-
ENGINE_NAME
- See Also:
-
-
Method Details
-
getAlternativeEngine
public ai.djl.engine.Engine getAlternativeEngine()- Specified by:
getAlternativeEngine
in classai.djl.engine.Engine
-
getEngineName
- Specified by:
getEngineName
in classai.djl.engine.Engine
-
getRank
public int getRank()- Specified by:
getRank
in classai.djl.engine.Engine
-
getVersion
- Specified by:
getVersion
in classai.djl.engine.Engine
-
hasCapability
- Specified by:
hasCapability
in classai.djl.engine.Engine
-
newModel
- Specified by:
newModel
in classai.djl.engine.Engine
-
newBaseManager
public ai.djl.ndarray.NDManager newBaseManager()- Specified by:
newBaseManager
in classai.djl.engine.Engine
-
newBaseManager
public ai.djl.ndarray.NDManager newBaseManager(ai.djl.Device device) - Specified by:
newBaseManager
in classai.djl.engine.Engine
-
toString
- Overrides:
toString
in classai.djl.engine.Engine
-