Package ai.djl.onnxruntime.engine

Class 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 
      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

      • Methods inherited from class java.lang.Object

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

    • Method Detail

      • getAlternativeEngine

        public ai.djl.engine.Engine getAlternativeEngine()
        Specified by:
        getAlternativeEngine in class ai.djl.engine.Engine

      • getEngineName

        public java.lang.String getEngineName()
        Specified by:
        getEngineName in class ai.djl.engine.Engine

      • getRank

        public int getRank()
        Specified by:
        getRank in class ai.djl.engine.Engine

      • getVersion

        public java.lang.String getVersion()
        Specified by:
        getVersion in class ai.djl.engine.Engine

      • hasCapability

        public boolean hasCapability​(java.lang.String capability)
        Specified by:
        hasCapability in class ai.djl.engine.Engine

      • newModel

        public ai.djl.Model newModel​(java.lang.String name,
                             ai.djl.Device device)
        Specified by:
        newModel in class ai.djl.engine.Engine

      • newBaseManager

        public ai.djl.ndarray.NDManager newBaseManager()
        Specified by:
        newBaseManager in class ai.djl.engine.Engine

      • newBaseManager

        public ai.djl.ndarray.NDManager newBaseManager​(ai.djl.Device device)
        Specified by:
        newBaseManager in class ai.djl.engine.Engine

      • toString

        public java.lang.String toString()
        Overrides:
        toString in class ai.djl.engine.Engine