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.training.GradientCollector newGradientCollector()
      ai.djl.Model newModel​(java.lang.String name, ai.djl.Device device)
      ai.djl.nn.SymbolBlock newSymbolBlock​(ai.djl.ndarray.NDManager manager)
      void setRandomSeed​(int seed)
      java.lang.String toString()

      • Methods inherited from class ai.djl.engine.Engine

        debugEnvironment, defaultDevice, getAllEngines, getDefaultEngineName, getDevices, getDevices, getDjlVersion, getEngine, getGpuCount, getInstance, getSeed, hasEngine, newParameterServer, registerEngine

      • 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

      • newSymbolBlock

        public ai.djl.nn.SymbolBlock newSymbolBlock​(ai.djl.ndarray.NDManager manager)
        Specified by:
        newSymbolBlock 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

      • newGradientCollector

        public ai.djl.training.GradientCollector newGradientCollector()
        Specified by:
        newGradientCollector in class ai.djl.engine.Engine

      • setRandomSeed

        public void setRandomSeed​(int seed)
        Overrides:
        setRandomSeed in class ai.djl.engine.Engine

      • toString

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