ai.djl.paddlepaddle.engine

Class PpSymbolBlock

java.lang.Object

ai.djl.nn.AbstractBlock

ai.djl.nn.AbstractSymbolBlock

ai.djl.paddlepaddle.engine.PpSymbolBlock

All Implemented Interfaces:

ai.djl.nn.Block, ai.djl.nn.SymbolBlock

public class PpSymbolBlock
extends ai.djl.nn.AbstractSymbolBlock

PpSymbolBlock is the PaddlePaddle implementation of SymbolBlock.

Field Summary

Fields inherited from class ai.djl.nn.AbstractBlock

children, inputShapes, parameters, version

Constructor Summary

Constructors

Constructor and Description
PpSymbolBlock(PaddlePredictor predictor, PpNDManager manager) Constructs a new PpSymbolBlock instance.

Method Summary

Modifier and Type	Method and Description
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)
ai.djl.ndarray.types.Shape[]	getOutputShapes(ai.djl.ndarray.types.Shape[] inputShapes)

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, removeLastBlock

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

PpSymbolBlock

public PpSymbolBlock(PaddlePredictor predictor,
                     PpNDManager manager)

Constructs a new PpSymbolBlock instance.

Parameters:

predictor - PaddlePredictor that holds the model information.

manager - the NDManager to holds the NDArray

Method Detail

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

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