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

    Fields inherited from interface ai.djl.ndarray.NDManager

    UID_GENERATOR

  • Method Summary

    Modifier and Type
    Method
    Description
    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(String data)
    ai.djl.ndarray.NDArray
    create(String[] data)
    ai.djl.ndarray.NDArray
    create(String[] data, Charset charset, ai.djl.ndarray.types.Shape shape)
    OrtNDArray
    create(Buffer data, ai.djl.ndarray.types.Shape shape, ai.djl.ndarray.types.DataType dataType)
    OrtNDArray
    from(ai.djl.ndarray.NDArray array)
    final 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 Details

    • allocateDirect

      public ByteBuffer allocateDirect(int capacity)

    • from

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

    • create

      public OrtNDArray create(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(String data)

    • create

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

    • create

      public ai.djl.ndarray.NDArray create(String[] data, 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 AutoCloseable
      Specified by:
      close in interface ai.djl.ndarray.NDManager
      Overrides:
      close in class ai.djl.ndarray.BaseNDManager