ai.djl.onnxruntime.engine

Class OrtNDManager

java.lang.Object

ai.djl.ndarray.BaseNDManager

ai.djl.onnxruntime.engine.OrtNDManager

All Implemented Interfaces:

ai.djl.ndarray.NDManager, java.lang.AutoCloseable

public class OrtNDManager
extends ai.djl.ndarray.BaseNDManager

OrtNDManager is the ONNX Runtime implementation of NDManager.

Nested Class Summary

Nested classes/interfaces inherited from class ai.djl.ndarray.BaseNDManager

ai.djl.ndarray.BaseNDManager.TempResource

Field Summary

Fields inherited from class ai.djl.ndarray.BaseNDManager

closed, device, name, parent, resources, tempResources, uid

Method Summary

Modifier and Type	Method and Description
java.nio.ByteBuffer	allocateDirect(int capacity)
void	close()
OrtNDArray	create(java.nio.Buffer data, ai.djl.ndarray.types.Shape shape, ai.djl.ndarray.types.DataType dataType)
ai.djl.ndarray.NDArray	create(java.lang.String data)
ai.djl.ndarray.NDArray	create(java.lang.String[] data)
ai.djl.ndarray.NDArray	create(java.lang.String[] data, java.nio.charset.Charset charset, ai.djl.ndarray.types.Shape shape)
OrtNDArray	from(ai.djl.ndarray.NDArray array)
ai.djl.engine.Engine	getEngine()
OrtNDManager	newSubManager(ai.djl.Device device)

Methods inherited from class ai.djl.ndarray.BaseNDManager

arange, attachInternal, copyBuffer, create, createCoo, createCSR, createRowSparse, debugDump, defaultDevice, detachInternal, eye, full, getAlternativeManager, getDevice, getName, getParentManager, invoke, invoke, isOpen, linspace, load, newSubManager, ones, randomInteger, randomMultinomial, randomMultinomial, randomNormal, randomUniform, setName, tempAttachInternal, toString, truncatedNormal, validateBufferSize, zeros

Methods inherited from class java.lang.Object

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

Methods inherited from interface ai.djl.ndarray.NDManager

arange, arange, arange, arange, arange, arange, arange, arange, attachAll, create, create, create, create, create, create, create, create, create, create, create, create, create, create, create, create, create, create, create, create, create, create, create, create, create, create, create, create, create, create, createCSR, createCSR, createRowSparse, decode, decode, eye, eye, eye, eye, full, full, full, linspace, linspace, linspace, linspace, load, newBaseManager, newBaseManager, newBaseManager, ones, ones, randomNormal, randomNormal, randomNormal, randomUniform, randomUniform, ret, subManagerOf, tempAttachAll, truncatedNormal, truncatedNormal, truncatedNormal, zeros, zeros

Method Detail

allocateDirect

public java.nio.ByteBuffer allocateDirect(int capacity)

from

public OrtNDArray from(ai.djl.ndarray.NDArray array)

create

public OrtNDArray create(java.nio.Buffer data,
                         ai.djl.ndarray.types.Shape shape,
                         ai.djl.ndarray.types.DataType dataType)

create

public ai.djl.ndarray.NDArray create(java.lang.String data)

create

public ai.djl.ndarray.NDArray create(java.lang.String[] data)

create

public ai.djl.ndarray.NDArray create(java.lang.String[] data,
                                     java.nio.charset.Charset charset,
                                     ai.djl.ndarray.types.Shape shape)

Specified by:

create in interface ai.djl.ndarray.NDManager

Overrides:

create in class ai.djl.ndarray.BaseNDManager

newSubManager

public OrtNDManager newSubManager(ai.djl.Device device)

getEngine

public final ai.djl.engine.Engine getEngine()

close

public void close()

Specified by:

close in interface ai.djl.ndarray.NDManager

Specified by:

close in interface java.lang.AutoCloseable

Overrides:

close in class ai.djl.ndarray.BaseNDManager