Index

A C D E F G H I L N O P R T
All Classes and Interfaces|All Packages|Constant Field Values

A

ai.djl.onnxruntime.engine - package ai.djl.onnxruntime.engine

Contains classes to interface with the underlying ONNXRuntime Engine.

ai.djl.onnxruntime.zoo - package ai.djl.onnxruntime.zoo

Contains the built-in OrtModelZoo.

ai.djl.onnxruntime.zoo.nlp.textgeneration - package ai.djl.onnxruntime.zoo.nlp.textgeneration

Contains classes for the Application.NLP.TEXT_GENERATION models.

ai.djl.onnxruntime.zoo.tabular.softmax_regression - package ai.djl.onnxruntime.zoo.tabular.softmax_regression

Contains classes for the classification models in the OrtModelZoo.

allocateDirect(int) - Method in class ai.djl.onnxruntime.engine.OrtNDManager

C

close() - Method in class ai.djl.onnxruntime.engine.OrtModel

close() - Method in class ai.djl.onnxruntime.engine.OrtNDArray

close() - Method in class ai.djl.onnxruntime.engine.OrtNDManager

close() - Method in class ai.djl.onnxruntime.engine.OrtSymbolBlock

create(boolean[]) - Method in class ai.djl.onnxruntime.engine.OrtNDManager

create(boolean[], Shape) - Method in class ai.djl.onnxruntime.engine.OrtNDManager

create(String) - Method in class ai.djl.onnxruntime.engine.OrtNDManager

create(String[]) - Method in class ai.djl.onnxruntime.engine.OrtNDManager

create(String[], Charset, Shape) - Method in class ai.djl.onnxruntime.engine.OrtNDManager

create(Buffer, Shape, DataType) - Method in class ai.djl.onnxruntime.engine.OrtNDManager

D

describeInput() - Method in class ai.djl.onnxruntime.engine.OrtSymbolBlock

detach() - Method in class ai.djl.onnxruntime.engine.OrtNDArray

E

ENGINE_NAME - Static variable in class ai.djl.onnxruntime.engine.OrtEngine

F

forwardInternal(ParameterStore, NDList, boolean, PairList<String, Object>) - Method in class ai.djl.onnxruntime.engine.OrtSymbolBlock

from(NDArray) - Method in class ai.djl.onnxruntime.engine.OrtNDManager

G

getAlternativeEngine() - Method in class ai.djl.onnxruntime.engine.OrtEngine

getDataType() - Method in class ai.djl.onnxruntime.engine.OrtNDArray

getDirectParameters() - Method in class ai.djl.onnxruntime.engine.OrtSymbolBlock

getEngine() - Method in class ai.djl.onnxruntime.engine.OrtEngineProvider

getEngine() - Method in class ai.djl.onnxruntime.engine.OrtNDManager

getEngineName() - Method in class ai.djl.onnxruntime.engine.OrtEngine

getEngineName() - Method in class ai.djl.onnxruntime.engine.OrtEngineProvider

getEngineRank() - Method in class ai.djl.onnxruntime.engine.OrtEngineProvider

getGroupId() - Method in class ai.djl.onnxruntime.zoo.OrtHfModelZoo

getGroupId() - Method in class ai.djl.onnxruntime.zoo.OrtModelZoo

getModelLoader(String) - Method in class ai.djl.onnxruntime.zoo.OrtHfModelZoo

getModelLoaders() - Method in class ai.djl.onnxruntime.zoo.OrtHfModelZoo

getModelZoo() - Method in class ai.djl.onnxruntime.zoo.OrtHfZooProvider

getModelZoo() - Method in class ai.djl.onnxruntime.zoo.OrtZooProvider

getPetalLength() - Method in class ai.djl.onnxruntime.zoo.tabular.softmax_regression.IrisFlower

Returns the petal length.

getPetalWidth() - Method in class ai.djl.onnxruntime.zoo.tabular.softmax_regression.IrisFlower

Returns the petal width.

getRank() - Method in class ai.djl.onnxruntime.engine.OrtEngine

getSepalLength() - Method in class ai.djl.onnxruntime.zoo.tabular.softmax_regression.IrisFlower

Returns the sepal length.

getSepalWidth() - Method in class ai.djl.onnxruntime.zoo.tabular.softmax_regression.IrisFlower

Returns the sepal width.

getShape() - Method in class ai.djl.onnxruntime.engine.OrtNDArray

getSupportedEngines() - Method in class ai.djl.onnxruntime.zoo.OrtHfModelZoo

getSupportedEngines() - Method in class ai.djl.onnxruntime.zoo.OrtModelZoo

getSupportedTypes() - Method in class ai.djl.onnxruntime.zoo.nlp.textgeneration.OrtGptTranslatorFactory

getSupportedTypes() - Method in class ai.djl.onnxruntime.zoo.tabular.softmax_regression.IrisClassificationTranslatorFactory

getTensor() - Method in class ai.djl.onnxruntime.engine.OrtNDArray

Returns the OnnxTensor representation of this OrtNDArray.

getVersion() - Method in class ai.djl.onnxruntime.engine.OrtEngine

GROUP_ID - Static variable in class ai.djl.onnxruntime.zoo.OrtModelZoo

H

hasCapability(String) - Method in class ai.djl.onnxruntime.engine.OrtEngine

I

intern(NDArray) - Method in class ai.djl.onnxruntime.engine.OrtNDArray

IrisClassificationTranslatorFactory - Class in ai.djl.onnxruntime.zoo.tabular.softmax_regression

A TranslatorFactory that creates a IrisClassificationTranslatorFactory.IrisTranslator instance.

IrisClassificationTranslatorFactory() - Constructor for class ai.djl.onnxruntime.zoo.tabular.softmax_regression.IrisClassificationTranslatorFactory

IrisFlower - Class in ai.djl.onnxruntime.zoo.tabular.softmax_regression

A class holds the iris flower features.

IrisFlower(float, float, float, float) - Constructor for class ai.djl.onnxruntime.zoo.tabular.softmax_regression.IrisFlower

Constructs a new IrisFlower instance.

L

load(InputStream, Map<String, ?>) - Method in class ai.djl.onnxruntime.engine.OrtModel

load(Path, String, Map<String, ?>) - Method in class ai.djl.onnxruntime.engine.OrtModel

N

newBaseManager() - Method in class ai.djl.onnxruntime.engine.OrtEngine

newBaseManager(Device) - Method in class ai.djl.onnxruntime.engine.OrtEngine

newInstance(Class<I>, Class<O>, Model, Map<String, ?>) - Method in class ai.djl.onnxruntime.zoo.nlp.textgeneration.OrtGptTranslatorFactory

newInstance(Class<I>, Class<O>, Model, Map<String, ?>) - Method in class ai.djl.onnxruntime.zoo.tabular.softmax_regression.IrisClassificationTranslatorFactory

newModel(String, Device) - Method in class ai.djl.onnxruntime.engine.OrtEngine

newSubManager(Device) - Method in class ai.djl.onnxruntime.engine.OrtNDManager

O

OrtEngine - Class in ai.djl.onnxruntime.engine

The OrtEngine is an implementation of the Engine based on the ONNX Runtime Deep Learning Library.

OrtEngineProvider - Class in ai.djl.onnxruntime.engine

OrtEngineProvider is the ONNX Runtime implementation of EngineProvider.

OrtEngineProvider() - Constructor for class ai.djl.onnxruntime.engine.OrtEngineProvider

OrtGptTranslator - Class in ai.djl.onnxruntime.zoo.nlp.textgeneration

The Translator for PyTorch GPT2 model.

OrtGptTranslator(long, int, int) - Constructor for class ai.djl.onnxruntime.zoo.nlp.textgeneration.OrtGptTranslator

Constructs a new instance of PtGptTranslator.

OrtGptTranslatorFactory - Class in ai.djl.onnxruntime.zoo.nlp.textgeneration

An TranslatorFactory that creates a OrtGptTranslator instance.

OrtGptTranslatorFactory() - Constructor for class ai.djl.onnxruntime.zoo.nlp.textgeneration.OrtGptTranslatorFactory

OrtHfModelZoo - Class in ai.djl.onnxruntime.zoo

OrtHfModelZoo is a repository that contains HuggingFace models for OnnxRuntime.

OrtHfZooProvider - Class in ai.djl.onnxruntime.zoo

An Huggingface model zoo provider for OnnxRuntime implements the ZooProvider interface.

OrtHfZooProvider() - Constructor for class ai.djl.onnxruntime.zoo.OrtHfZooProvider

OrtModel - Class in ai.djl.onnxruntime.engine

OrtModel is the ONNX Runtime implementation of Model.

OrtModelZoo - Class in ai.djl.onnxruntime.zoo

OrtModelZoo is a repository that contains all Onnx models for DJL.

OrtNDArray - Class in ai.djl.onnxruntime.engine

OrtNDArray is the ONNX Runtime implementation of NDArray.

OrtNDManager - Class in ai.djl.onnxruntime.engine

OrtNDManager is the ONNX Runtime implementation of NDManager.

OrtSymbolBlock - Class in ai.djl.onnxruntime.engine

OrtSymbolBlock is the ONNX Runtime implementation of SymbolBlock.

OrtSymbolBlock(OrtSession, OrtNDManager) - Constructor for class ai.djl.onnxruntime.engine.OrtSymbolBlock

Constructs a OrtSymbolBlock.

OrtZooProvider - Class in ai.djl.onnxruntime.zoo

An Onnx Runtime model zoo provider implements the ZooProvider interface.

OrtZooProvider() - Constructor for class ai.djl.onnxruntime.zoo.OrtZooProvider

P

processInput(TranslatorContext, NDList) - Method in class ai.djl.onnxruntime.zoo.nlp.textgeneration.OrtGptTranslator

processOutput(TranslatorContext, NDList) - Method in class ai.djl.onnxruntime.zoo.nlp.textgeneration.OrtGptTranslator

R

removeLastBlock() - Method in class ai.djl.onnxruntime.engine.OrtSymbolBlock

T

toByteBuffer(boolean) - Method in class ai.djl.onnxruntime.engine.OrtNDArray

toString() - Method in class ai.djl.onnxruntime.engine.OrtEngine

toStringArray(Charset) - Method in class ai.djl.onnxruntime.engine.OrtNDArray

A C D E F G H I L N O P R T
All Classes and Interfaces|All Packages|Constant Field Values