ai.djl.tensorflow.engine

Class TfSymbolBlock

java.lang.Object

ai.djl.tensorflow.engine.TfSymbolBlock

All Implemented Interfaces:

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

public class TfSymbolBlock
extends java.lang.Object
implements ai.djl.nn.SymbolBlock

Constructor Summary

Constructors

Constructor and Description
TfSymbolBlock(org.tensorflow.SavedModelBundle bundle)

Method Summary

Modifier and Type	Method and Description
void	cast(ai.djl.ndarray.types.DataType dataType)
void	clear()
ai.djl.util.PairList<java.lang.String,ai.djl.ndarray.types.Shape>	describeInput()
ai.djl.ndarray.NDList	forward(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.nn.BlockList	getChildren()
ai.djl.nn.ParameterList	getDirectParameters()
ai.djl.ndarray.types.Shape[]	getOutputShapes(ai.djl.ndarray.NDManager manager, ai.djl.ndarray.types.Shape[] inputShapes)
ai.djl.nn.ParameterList	getParameters()
ai.djl.ndarray.types.Shape	getParameterShape(java.lang.String name, ai.djl.ndarray.types.Shape[] inputShapes)
ai.djl.ndarray.types.Shape[]	initialize(ai.djl.ndarray.NDManager manager, ai.djl.ndarray.types.DataType dataType, ai.djl.ndarray.types.Shape... inputShapes)
boolean	isInitialized()
void	loadParameters(ai.djl.ndarray.NDManager manager, java.io.DataInputStream is)
void	removeLastBlock()
void	saveParameters(java.io.DataOutputStream os)
void	setInitializer(ai.djl.training.initializer.Initializer initializer)
void	setInitializer(ai.djl.training.initializer.Initializer initializer, java.lang.String paramName)

Methods inherited from class java.lang.Object

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

Methods inherited from interface ai.djl.nn.Block

forward, forward, validateLayout

Constructor Detail

TfSymbolBlock

public TfSymbolBlock(org.tensorflow.SavedModelBundle bundle)

Method Detail

removeLastBlock

public void removeLastBlock()

Specified by:

removeLastBlock in interface ai.djl.nn.SymbolBlock

forward

public ai.djl.ndarray.NDList forward(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:

forward in interface ai.djl.nn.Block

setInitializer

public void setInitializer(ai.djl.training.initializer.Initializer initializer)

Specified by:

setInitializer in interface ai.djl.nn.Block

setInitializer

public void setInitializer(ai.djl.training.initializer.Initializer initializer,
                           java.lang.String paramName)

Specified by:

setInitializer in interface ai.djl.nn.Block

initialize

public ai.djl.ndarray.types.Shape[] 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

isInitialized

public boolean isInitialized()

Specified by:

isInitialized in interface ai.djl.nn.Block

cast

public void cast(ai.djl.ndarray.types.DataType dataType)

Specified by:

cast in interface ai.djl.nn.Block

clear

public void clear()

Specified by:

clear in interface ai.djl.nn.Block

describeInput

public ai.djl.util.PairList<java.lang.String,ai.djl.ndarray.types.Shape> describeInput()

Specified by:

describeInput in interface ai.djl.nn.Block

getChildren

public ai.djl.nn.BlockList getChildren()

Specified by:

getChildren in interface ai.djl.nn.Block

getDirectParameters

public ai.djl.nn.ParameterList getDirectParameters()

Specified by:

getDirectParameters in interface ai.djl.nn.Block

getParameters

public ai.djl.nn.ParameterList getParameters()

Specified by:

getParameters in interface ai.djl.nn.Block

getParameterShape

public ai.djl.ndarray.types.Shape getParameterShape(java.lang.String name,
                                                    ai.djl.ndarray.types.Shape[] inputShapes)

Specified by:

getParameterShape in interface ai.djl.nn.Block

getOutputShapes

public ai.djl.ndarray.types.Shape[] getOutputShapes(ai.djl.ndarray.NDManager manager,
                                                    ai.djl.ndarray.types.Shape[] inputShapes)

Specified by:

getOutputShapes in interface ai.djl.nn.Block

saveParameters

public void saveParameters(java.io.DataOutputStream os)

Specified by:

saveParameters in interface ai.djl.nn.Block

loadParameters

public void loadParameters(ai.djl.ndarray.NDManager manager,
                           java.io.DataInputStream is)

Specified by:

loadParameters in interface ai.djl.nn.Block