Package ai.djl.onnxruntime.engine

Class OrtSymbolBlock

java.lang.Object
ai.djl.nn.AbstractBaseBlock
ai.djl.nn.AbstractSymbolBlock
ai.djl.onnxruntime.engine.OrtSymbolBlock
All Implemented Interfaces:
ai.djl.nn.Block, ai.djl.nn.SymbolBlock, AutoCloseable
public class OrtSymbolBlock extends ai.djl.nn.AbstractSymbolBlock implements AutoCloseable
OrtSymbolBlock is the ONNX Runtime implementation of SymbolBlock.

You can create a OrtSymbolBlock using Model.load(java.nio.file.Path, String).

  • Field Summary

    Fields inherited from class ai.djl.nn.AbstractBaseBlock

    inputNames, inputShapes, outputDataTypes, version

  • Constructor Summary

    Constructors
    Constructor
    Description
    OrtSymbolBlock(ai.onnxruntime.OrtSession session, OrtNDManager manager)
    Constructs a OrtSymbolBlock.

  • Method Summary

    Modifier and Type
    Method
    Description
    void
    close()
    ai.djl.util.PairList<String,ai.djl.ndarray.types.Shape>
    describeInput()
    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)
    Map<String,String>
    getCustomMetadata()
    ai.djl.nn.ParameterList
    getDirectParameters()
    void
    removeLastBlock()

    Methods inherited from class ai.djl.nn.AbstractSymbolBlock

    getChildren, getOutputShapes

    Methods inherited from class ai.djl.nn.AbstractBaseBlock

    beforeInitialize, cast, clear, forward, forward, forwardInternal, getInputShapes, getOutputDataTypes, getParameters, initialize, initializeChildBlocks, isInitialized, 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, getInputShapes, getOutputDataTypes, getOutputShapes, getParameters, initialize, isInitialized, loadParameters, saveParameters, setInitializer, setInitializer, setInitializer

    Methods inherited from interface ai.djl.nn.SymbolBlock

    describeOutput

  • Constructor Details

    • OrtSymbolBlock

      public OrtSymbolBlock(ai.onnxruntime.OrtSession session, OrtNDManager manager)
      Constructs a OrtSymbolBlock.

      You can create a PtSymbolBlock using Model.load(java.nio.file.Path, String).

      Parameters:
      session - the OrtSession contains the model information
      manager - the NDManager to holds the NDArray

  • Method Details

    • 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

    • describeInput

      public 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

    • getCustomMetadata

      public Map<String,String> getCustomMetadata()
      Specified by:
      getCustomMetadata in interface ai.djl.nn.Block

    • close

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

    • getDirectParameters

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