Package ai.djl.tensorflow.engine

Class TfEngine

java.lang.Object

ai.djl.engine.Engine

ai.djl.tensorflow.engine.TfEngine

All Implemented Interfaces:

AutoCloseable

public final class TfEngine
extends ai.djl.engine.Engine
implements AutoCloseable

The TfEngine is an implementation of the Engine based on the Tensorflow Deep Learning Framework.

To get an instance of the TfEngine when it is not the default Engine, call Engine.getEngine(String) with the Engine name "TensorFlow".

Field Summary

Fields

Modifier and Type	Field	Description
static final String	ENGINE_NAME

Method Summary

Modifier and Type	Method	Description
void	close()
ai.djl.engine.Engine	getAlternativeEngine()
String	getEngineName()
int	getRank()
String	getVersion()
boolean	hasCapability(String capability)
ai.djl.ndarray.NDManager	newBaseManager()
ai.djl.ndarray.NDManager	newBaseManager(ai.djl.Device device)
ai.djl.Model	newModel(String name, ai.djl.Device device)
ai.djl.nn.SymbolBlock	newSymbolBlock(ai.djl.ndarray.NDManager manager)
String	toString()

Methods inherited from class ai.djl.engine.Engine

debugEnvironment, defaultDevice, getAllEngines, getDefaultEngine, getDefaultEngineName, getDevices, getDevices, getDjlVersion, getEngine, getGpuCount, getInstance, getSeed, hasEngine, newGradientCollector, newParameterServer, registerEngine, setDefaultEngine, setRandomSeed

Methods inherited from class java.lang.Object

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

Field Details

ENGINE_NAME

public static final String ENGINE_NAME

See Also:

• Constant Field Values

Method Details

newModel

public ai.djl.Model newModel(String name,
 ai.djl.Device device)

Specified by:

newModel in class ai.djl.engine.Engine

newSymbolBlock

public ai.djl.nn.SymbolBlock newSymbolBlock(ai.djl.ndarray.NDManager manager)

Overrides:

newSymbolBlock in class ai.djl.engine.Engine

getAlternativeEngine

public ai.djl.engine.Engine getAlternativeEngine()

Specified by:

getAlternativeEngine in class ai.djl.engine.Engine

getEngineName

public String getEngineName()

Specified by:

getEngineName in class ai.djl.engine.Engine

getRank

public int getRank()

Specified by:

getRank in class ai.djl.engine.Engine

getVersion

public String getVersion()

Specified by:

getVersion in class ai.djl.engine.Engine

hasCapability

public boolean hasCapability(String capability)

Specified by:

hasCapability in class ai.djl.engine.Engine

newBaseManager

public ai.djl.ndarray.NDManager newBaseManager()

Specified by:

newBaseManager in class ai.djl.engine.Engine

newBaseManager

public ai.djl.ndarray.NDManager newBaseManager(ai.djl.Device device)

Specified by:

newBaseManager in class ai.djl.engine.Engine

toString

public String toString()

Overrides:

toString in class ai.djl.engine.Engine

close

public void close()

Specified by:

close in interface AutoCloseable