Package ai.djl.python.engine

Class PyEngine

  • public final class PyEngine
extends ai.djl.engine.Engine
    The PyEngine is an implementation of the Engine that runs Python worker.

    • 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)

      • 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, toString

      • 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