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
,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 TypeMethodDescriptionallocateDirect
(int capacity) void
close()
create
(boolean[] data) create
(boolean[] data, ai.djl.ndarray.types.Shape shape) ai.djl.ndarray.NDArray
ai.djl.ndarray.NDArray
ai.djl.ndarray.NDArray
from
(ai.djl.ndarray.NDArray array) final ai.djl.engine.Engine
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, getManagedArrays, getName, getParentManager, invoke, invoke, isOpen, linspace, load, newSubManager, randomInteger, randomMultinomial, randomMultinomial, randomNormal, randomPermutation, randomUniform, sampleGamma, sampleGamma, sampleNormal, sampleNormal, samplePoisson, samplePoisson, setName, tempAttachInternal, toString, truncatedNormal, validateBuffer
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, createCSR, createCSR, createRowSparse, decode, decode, eye, eye, eye, eye, full, full, full, hanningWindow, linspace, linspace, linspace, linspace, load, ones, ones, ones, randomNormal, randomNormal, randomNormal, randomUniform, randomUniform, ret, tempAttachAll, truncatedNormal, truncatedNormal, truncatedNormal, zeros, zeros, zeros
-
Method Details
-
allocateDirect
-
from
-
create
public OrtNDArray create(Buffer data, ai.djl.ndarray.types.Shape shape, ai.djl.ndarray.types.DataType dataType) -
create
-
create
-
create
-
create
-
create
public ai.djl.ndarray.NDArray create(String[] data, Charset charset, ai.djl.ndarray.types.Shape shape) - Specified by:
create
in interfaceai.djl.ndarray.NDManager
- Overrides:
create
in classai.djl.ndarray.BaseNDManager
-
newSubManager
-
getEngine
public final ai.djl.engine.Engine getEngine() -
close
public void close()- Specified by:
close
in interfaceAutoCloseable
- Specified by:
close
in interfaceai.djl.ndarray.NDManager
- Overrides:
close
in classai.djl.ndarray.BaseNDManager
-