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.EngineProvider

PtEngineProvider is the PyTorch implementation of EngineProvider.

Constructor Summary

Constructors

Constructor and Description
PtEngineProvider()

Method Summary

Modifier and Type	Method and Description
ai.djl.engine.Engine	getEngine()
java.lang.String	getEngineName()
int	getEngineRank()

Methods inherited from class java.lang.Object

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

Constructor Detail

PtEngineProvider

public PtEngineProvider()

Method Detail

getEngineName

public java.lang.String getEngineName()

Specified by:

getEngineName in interface ai.djl.engine.EngineProvider

getEngineRank

public int getEngineRank()

Specified by:

getEngineRank in interface ai.djl.engine.EngineProvider

getEngine

public ai.djl.engine.Engine getEngine()

Specified by:

getEngine in interface ai.djl.engine.EngineProvider