public final class TfEngine
extends ai.djl.engine.Engine
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".
| Modifier and Type | Field and Description |
|---|---|
static java.lang.String |
ENGINE_NAME |
| Modifier and Type | Method and Description |
|---|---|
java.lang.String |
getEngineName() |
java.lang.String |
getVersion() |
boolean |
hasCapability(java.lang.String capability) |
ai.djl.ndarray.NDManager |
newBaseManager() |
ai.djl.ndarray.NDManager |
newBaseManager(ai.djl.Device device) |
ai.djl.training.GradientCollector |
newGradientCollector() |
ai.djl.Model |
newModel(java.lang.String name,
ai.djl.Device device) |
void |
setRandomSeed(int seed) |
public static final java.lang.String ENGINE_NAME
public ai.djl.Model newModel(java.lang.String name,
ai.djl.Device device)
newModel in class ai.djl.engine.Enginepublic java.lang.String getEngineName()
getEngineName in class ai.djl.engine.Enginepublic java.lang.String getVersion()
getVersion in class ai.djl.engine.Enginepublic boolean hasCapability(java.lang.String capability)
hasCapability in class ai.djl.engine.Enginepublic ai.djl.ndarray.NDManager newBaseManager()
newBaseManager in class ai.djl.engine.Enginepublic ai.djl.ndarray.NDManager newBaseManager(ai.djl.Device device)
newBaseManager in class ai.djl.engine.Enginepublic ai.djl.training.GradientCollector newGradientCollector()
newGradientCollector in class ai.djl.engine.Enginepublic void setRandomSeed(int seed)
setRandomSeed in class ai.djl.engine.Engine