ai.djl.tensorflow.engine

Class TfModel

java.lang.Object

ai.djl.BaseModel

ai.djl.tensorflow.engine.TfModel

All Implemented Interfaces:

ai.djl.Model, java.lang.AutoCloseable

public class TfModel
extends ai.djl.BaseModel

Field Summary

Fields inherited from class ai.djl.BaseModel

artifacts, block, dataType, inputData, modelDir, modelName, properties

Method Summary

Modifier and Type	Method and Description
void	cast(ai.djl.ndarray.types.DataType dataType)
void	close()
java.lang.String[]	getArtifactNames()
ai.djl.nn.Block	getBlock()
ai.djl.ndarray.NDManager	getNDManager()
void	load(java.nio.file.Path modelPath, java.lang.String prefix, java.util.Map<java.lang.String,java.lang.Object> options)
<I,O> ai.djl.inference.Predictor<I,O>	newPredictor(ai.djl.translate.Translator<I,O> translator)
ai.djl.training.Trainer	newTrainer(ai.djl.training.TrainingConfig trainingConfig)
void	save(java.nio.file.Path modelPath, java.lang.String newModelName)
void	setBlock(ai.djl.nn.Block block)

Methods inherited from class ai.djl.BaseModel

describeInput, describeOutput, finalize, getArtifact, getArtifact, getArtifactAsStream, getDataType, getModelPath, getName, getProperty, paramPathResolver, readParameters, setDataType, setModelDir, setProperty

Methods inherited from class java.lang.Object

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

Methods inherited from interface ai.djl.Model

load, load, newInstance, newInstance, newInstance, newInstance, quantize

Method Detail

load

public void load(java.nio.file.Path modelPath,
                 java.lang.String prefix,
                 java.util.Map<java.lang.String,java.lang.Object> options)
          throws java.io.FileNotFoundException

Throws:

java.io.FileNotFoundException

save

public void save(java.nio.file.Path modelPath,
                 java.lang.String newModelName)

Specified by:

save in interface ai.djl.Model

Overrides:

save in class ai.djl.BaseModel

getBlock

public ai.djl.nn.Block getBlock()

Specified by:

getBlock in interface ai.djl.Model

Overrides:

getBlock in class ai.djl.BaseModel

setBlock

public void setBlock(ai.djl.nn.Block block)

Specified by:

setBlock in interface ai.djl.Model

Overrides:

setBlock in class ai.djl.BaseModel

newTrainer

public ai.djl.training.Trainer newTrainer(ai.djl.training.TrainingConfig trainingConfig)

newPredictor

public <I,O> ai.djl.inference.Predictor<I,O> newPredictor(ai.djl.translate.Translator<I,O> translator)

getNDManager

public ai.djl.ndarray.NDManager getNDManager()

Specified by:

getNDManager in interface ai.djl.Model

Overrides:

getNDManager in class ai.djl.BaseModel

getArtifactNames

public java.lang.String[] getArtifactNames()

cast

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

close

public void close()