public class OrtNDManager
extends ai.djl.ndarray.BaseNDManager
OrtNDManager is the ONNX Runtime implementation of NDManager.| 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) |
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, zerosclone, equals, finalize, getClass, hashCode, notify, notifyAll, wait, wait, waitarange, 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, zerospublic java.nio.ByteBuffer allocateDirect(int capacity)
public OrtNDArray from(ai.djl.ndarray.NDArray array)
public OrtNDArray create(java.nio.Buffer data, ai.djl.ndarray.types.Shape shape, ai.djl.ndarray.types.DataType dataType)
public ai.djl.ndarray.NDArray create(java.lang.String data)
public ai.djl.ndarray.NDArray create(java.lang.String[] data)
public ai.djl.ndarray.NDArray create(java.lang.String[] data,
java.nio.charset.Charset charset,
ai.djl.ndarray.types.Shape shape)
create in interface ai.djl.ndarray.NDManagercreate in class ai.djl.ndarray.BaseNDManagerpublic OrtNDManager newSubManager(ai.djl.Device device)
public final ai.djl.engine.Engine getEngine()
public void close()
close in interface ai.djl.ndarray.NDManagerclose in interface java.lang.AutoCloseableclose in class ai.djl.ndarray.BaseNDManager