Package ai.djl.pytorch.engine
Class PtEngineProvider
- java.lang.Object
-
- ai.djl.pytorch.engine.PtEngineProvider
-
- All Implemented Interfaces:
ai.djl.engine.EngineProvider
public class PtEngineProvider extends java.lang.Object implements ai.djl.engine.EngineProviderPtEngineProvideris the PyTorch implementation ofEngineProvider.
-
-
Constructor Summary
Constructors Constructor Description PtEngineProvider()
-
Method Summary
All Methods Instance Methods Concrete Methods Modifier and Type Method Description ai.djl.engine.EnginegetEngine()java.lang.StringgetEngineName()intgetEngineRank()
-
-
-
Method Detail
-
getEngineName
public java.lang.String getEngineName()
- Specified by:
getEngineNamein interfaceai.djl.engine.EngineProvider
-
getEngineRank
public int getEngineRank()
- Specified by:
getEngineRankin interfaceai.djl.engine.EngineProvider
-
getEngine
public ai.djl.engine.Engine getEngine()
- Specified by:
getEnginein interfaceai.djl.engine.EngineProvider
-
-