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()

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

getEngine

public ai.djl.engine.Engine getEngine()

Specified by:

getEngine in interface ai.djl.engine.EngineProvider