Package ai.djl.tensorflow.engine

Class TfSymbolBlock

java.lang.Object
ai.djl.nn.AbstractBaseBlock
ai.djl.nn.AbstractSymbolBlock
ai.djl.tensorflow.engine.TfSymbolBlock
All Implemented Interfaces:
ai.djl.nn.Block, ai.djl.nn.SymbolBlock, AutoCloseable
public class TfSymbolBlock extends ai.djl.nn.AbstractSymbolBlock implements AutoCloseable
TfSymbolBlock is the TensorFlow implementation of SymbolBlock.

  • Field Summary

    Fields inherited from class ai.djl.nn.AbstractBaseBlock

    inputNames, inputShapes, outputDataTypes, version

  • Constructor Summary

    Constructors
    Constructor
    Description
    TfSymbolBlock(SavedModelBundle bundle, String signatureDefKey)
     

  • Method Summary

    Modifier and Type
    Method
    Description
    void
    close()
    final ai.djl.util.PairList<String,ai.djl.ndarray.types.Shape>
    describeInput()
    final ai.djl.util.PairList<String,ai.djl.ndarray.types.Shape>
    describeOutput()
    protected ai.djl.ndarray.NDList
    forwardInternal(ai.djl.training.ParameterStore parameterStore, ai.djl.ndarray.NDList inputs, boolean training, ai.djl.util.PairList<String,Object> params)
    ai.djl.nn.ParameterList
    getDirectParameters()
    ai.djl.ndarray.types.Shape[]
    getOutputShapes(ai.djl.ndarray.types.Shape[] inputShapes)
    SavedModelBundle
    getSavedModelBundle()
    Returns the SavedModelBundle for the model.
    void
    initialize(ai.djl.ndarray.NDManager manager, ai.djl.ndarray.types.DataType dataType, ai.djl.ndarray.types.Shape... inputShapes)
    boolean
    isInitialized()
    void
    removeLastBlock()

    Methods inherited from class ai.djl.nn.AbstractSymbolBlock

    getChildren

    Methods inherited from class ai.djl.nn.AbstractBaseBlock

    beforeInitialize, cast, clear, forward, forward, forwardInternal, getInputShapes, getOutputDataTypes, getParameters, initializeChildBlocks, loadMetadata, loadParameters, prepare, readInputShapes, saveInputShapes, saveMetadata, saveParameters, setInitializer, setInitializer, setInitializer, toString

    Methods inherited from class java.lang.Object

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

    Methods inherited from interface ai.djl.nn.Block

    cast, clear, forward, forward, forward, freezeParameters, freezeParameters, getCustomMetadata, getInputShapes, getOutputDataTypes, getOutputShapes, getParameters, loadParameters, saveParameters, setInitializer, setInitializer, setInitializer

  • Constructor Details

  • Method Details

    • getSavedModelBundle

      public SavedModelBundle getSavedModelBundle()
      Returns the SavedModelBundle for the model.
      Returns:
      the SavedModelBundle for the model

    • removeLastBlock

      public void removeLastBlock()
      Specified by:
      removeLastBlock in interface ai.djl.nn.SymbolBlock

    • forwardInternal

      protected ai.djl.ndarray.NDList forwardInternal(ai.djl.training.ParameterStore parameterStore, ai.djl.ndarray.NDList inputs, boolean training, ai.djl.util.PairList<String,Object> params)
      Specified by:
      forwardInternal in class ai.djl.nn.AbstractBaseBlock

    • initialize

      public void initialize(ai.djl.ndarray.NDManager manager, ai.djl.ndarray.types.DataType dataType, ai.djl.ndarray.types.Shape... inputShapes)
      Specified by:
      initialize in interface ai.djl.nn.Block
      Overrides:
      initialize in class ai.djl.nn.AbstractBaseBlock

    • isInitialized

      public boolean isInitialized()
      Specified by:
      isInitialized in interface ai.djl.nn.Block
      Overrides:
      isInitialized in class ai.djl.nn.AbstractBaseBlock

    • describeInput

      public final ai.djl.util.PairList<String,ai.djl.ndarray.types.Shape> describeInput()
      Specified by:
      describeInput in interface ai.djl.nn.Block
      Overrides:
      describeInput in class ai.djl.nn.AbstractBaseBlock

    • getDirectParameters

      public ai.djl.nn.ParameterList getDirectParameters()
      Specified by:
      getDirectParameters in interface ai.djl.nn.Block

    • describeOutput

      public final ai.djl.util.PairList<String,ai.djl.ndarray.types.Shape> describeOutput()
      Specified by:
      describeOutput in interface ai.djl.nn.SymbolBlock

    • getOutputShapes

      public ai.djl.ndarray.types.Shape[] getOutputShapes(ai.djl.ndarray.types.Shape[] inputShapes)
      Specified by:
      getOutputShapes in interface ai.djl.nn.Block
      Overrides:
      getOutputShapes in class ai.djl.nn.AbstractSymbolBlock

    • close

      public void close()
      Specified by:
      close in interface AutoCloseable