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 ofNDManager
.
-
-
Method Summary
All Methods Instance Methods Concrete Methods Modifier and Type Method Description java.nio.ByteBuffer
allocateDirect(int capacity)
void
close()
OrtNDArray
create(boolean[] data)
OrtNDArray
create(boolean[] data, ai.djl.ndarray.types.Shape shape)
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, 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 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 OrtNDArray create(boolean[] data)
-
create
public OrtNDArray create(boolean[] data, ai.djl.ndarray.types.Shape shape)
-
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 interfaceai.djl.ndarray.NDManager
- Overrides:
create
in classai.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 interfacejava.lang.AutoCloseable
- Specified by:
close
in interfaceai.djl.ndarray.NDManager
- Overrides:
close
in classai.djl.ndarray.BaseNDManager
-
-