ai.djl.onnxruntime.engine

Class OrtSymbolBlock

java.lang.Object

ai.djl.nn.AbstractBlock

ai.djl.nn.AbstractSymbolBlock

ai.djl.onnxruntime.engine.OrtSymbolBlock

All Implemented Interfaces:

ai.djl.nn.Block, ai.djl.nn.SymbolBlock, java.lang.AutoCloseable

public class OrtSymbolBlock
extends ai.djl.nn.AbstractSymbolBlock
implements java.lang.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.AbstractBlock

children, inputNames, inputShapes, parameters, version

Constructor Summary

Constructors

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

Method Summary

Modifier and Type	Method and Description
void	close()
protected ai.djl.ndarray.NDList	forwardInternal(ai.djl.training.ParameterStore parameterStore, ai.djl.ndarray.NDList inputs, boolean training, ai.djl.util.PairList<java.lang.String,java.lang.Object> params)
void	removeLastBlock()

Methods inherited from class ai.djl.nn.AbstractSymbolBlock

getOutputShapes

Methods inherited from class ai.djl.nn.AbstractBlock

addChildBlock, addParameter, beforeInitialize, cast, clear, describeInput, forward, forward, forwardInternal, getChildren, getDirectParameters, 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.SymbolBlock

describeOutput, newInstance

Methods inherited from interface ai.djl.nn.Block

cast, clear, describeInput, forward, forward, forward, getChildren, getDirectParameters, getParameters, initialize, isInitialized, loadParameters, saveParameters, setInitializer, setInitializer, setInitializer, validateLayout

Constructor Detail

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 Detail

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<java.lang.String,java.lang.Object> params)

Specified by:

forwardInternal in class ai.djl.nn.AbstractBlock

close

public void close()

Specified by:

close in interface java.lang.AutoCloseable