PtEngineProvider (Deep Java Library 0.35.0 - pytorch-engine API)

java.lang.Object

ai.djl.pytorch.engine.PtEngineProvider

All Implemented Interfaces:: ai.djl.engine.EngineProvider

public class PtEngineProvider extends Object implements ai.djl.engine.EngineProvider

PtEngineProvider is the PyTorch implementation of EngineProvider.

Constructor Summary

Constructors

Constructor

Description

PtEngineProvider()
Method Summary

Modifier and Type

Method

Description

ai.djl.engine.Engine

getEngine()

String

getEngineName()

int

getEngineRank()

Methods inherited from class java.lang.Object
clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Details
- PtEngineProvider
  
  public PtEngineProvider()
Method Details
- getEngineName
  
  public 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