Package 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

Nested classes/interfaces inherited from interface ai.djl.ndarray.NDManager

ai.djl.ndarray.NDManager.SystemNDManager

Field Summary

Fields inherited from class ai.djl.ndarray.BaseNDManager

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

Method Summary

Modifier and Type	Method	Description
java.nio.ByteBuffer	allocateDirect(int capacity)
void	close()
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	create(java.nio.Buffer data, ai.djl.ndarray.types.Shape shape, ai.djl.ndarray.types.DataType dataType)
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, attachUncappedInternal, cap, copyBuffer, create, createCoo, createCSR, createRowSparse, debugDump, defaultDevice, detachInternal, eye, full, getDevice, getName, getParentManager, invoke, invoke, isOpen, linspace, load, newSubManager, ones, randomInteger, randomMultinomial, randomMultinomial, randomNormal, randomUniform, sampleGamma, sampleGamma, sampleNormal, sampleNormal, samplePoisson, samplePoisson, setName, tempAttachInternal, toString, truncatedNormal, validateBuffer, 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, ones, ones, randomNormal, randomNormal, randomNormal, randomUniform, randomUniform, ret, 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 java.lang.AutoCloseable

Specified by:

close in interface ai.djl.ndarray.NDManager

Overrides:

close in class ai.djl.ndarray.BaseNDManager