Package ai.djl.python.engine

Class PyEngine

java.lang.Object

ai.djl.engine.Engine

ai.djl.python.engine.PyEngine

public final class PyEngine
extends ai.djl.engine.Engine

The PyEngine is an implementation of the Engine that runs Python worker.

Method Summary

Modifier and Type	Method	Description
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)

Methods inherited from class ai.djl.engine.Engine

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

Methods inherited from class java.lang.Object

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

Method Details

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

newModel

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

Specified by:

newModel 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