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
TheOrtEngine
is an implementation of theEngine
based on the ONNX Runtime Deep Learning Library.To get an instance of the
OrtEngine
when it is not the default Engine, callEngine.getEngine(String)
with the Engine name "OnnxRuntime".
-
-
Field Summary
Fields Modifier and Type Field Description static java.lang.String
ENGINE_NAME
-
Method Summary
All Methods Instance Methods Concrete Methods Modifier and Type Method Description ai.djl.engine.Engine
getAlternativeEngine()
java.lang.String
getEngineName()
int
getRank()
java.lang.String
getVersion()
boolean
hasCapability(java.lang.String capability)
ai.djl.ndarray.NDManager
newBaseManager()
ai.djl.ndarray.NDManager
newBaseManager(ai.djl.Device device)
ai.djl.Model
newModel(java.lang.String name, ai.djl.Device device)
java.lang.String
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 Detail
-
ENGINE_NAME
public static final java.lang.String ENGINE_NAME
- See Also:
- Constant Field Values
-
-
Method Detail
-
getAlternativeEngine
public ai.djl.engine.Engine getAlternativeEngine()
- Specified by:
getAlternativeEngine
in classai.djl.engine.Engine
-
getEngineName
public java.lang.String getEngineName()
- Specified by:
getEngineName
in classai.djl.engine.Engine
-
getRank
public int getRank()
- Specified by:
getRank
in classai.djl.engine.Engine
-
getVersion
public java.lang.String getVersion()
- Specified by:
getVersion
in classai.djl.engine.Engine
-
hasCapability
public boolean hasCapability(java.lang.String capability)
- Specified by:
hasCapability
in classai.djl.engine.Engine
-
newModel
public ai.djl.Model newModel(java.lang.String name, ai.djl.Device device)
- 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
public java.lang.String toString()
- Overrides:
toString
in classai.djl.engine.Engine
-
-