Package ai.djl.tensorflow.engine

Class TfEngine

  • All Implemented Interfaces:
    java.lang.AutoCloseable
    public final class TfEngine
extends ai.djl.engine.Engine
implements java.lang.AutoCloseable
    The TfEngine is an implementation of the Engine based on the Tensorflow Deep Learning Framework.

    To get an instance of the TfEngine when it is not the default Engine, call Engine.getEngine(String) with the Engine name "TensorFlow".

    • 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
      void close()
      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)
      ai.djl.nn.SymbolBlock newSymbolBlock​(ai.djl.ndarray.NDManager manager)
      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, registerEngine, setRandomSeed

      • Methods inherited from class java.lang.Object

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

    • Method Detail

      • 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)
        Overrides:
        newSymbolBlock in class ai.djl.engine.Engine

      • 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

      • 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

      • close

        public void close()
        Specified by:
        close in interface java.lang.AutoCloseable