Package ai.djl.paddlepaddle.jni

Class JniUtils

java.lang.Object
ai.djl.paddlepaddle.jni.JniUtils
public final class JniUtils extends Object
A class containing utilities to interact with the Paddle Engine's Java Native Interface (JNI) layer.

  • Method Details

    • createNdArray

      public static PpNDArray createNdArray(PpNDManager manager, ByteBuffer data, ai.djl.ndarray.types.Shape shape, ai.djl.ndarray.types.DataType dtype)

    • getDTypeFromNd

      public static ai.djl.ndarray.types.DataType getDTypeFromNd(PpNDArray array)

    • getByteBufferFromNd

      public static ByteBuffer getByteBufferFromNd(PpNDArray array)

    • getShapeFromNd

      public static ai.djl.ndarray.types.Shape getShapeFromNd(PpNDArray array)

    • setNdName

      public static void setNdName(PpNDArray array, String name)

    • getNameFromNd

      public static String getNameFromNd(PpNDArray array)

    • setNdLoD

      public static void setNdLoD(PpNDArray array, long[][] lod)

    • getNdLoD

      public static long[][] getNdLoD(PpNDArray array)

    • deleteNd

      public static void deleteNd(Long handle)

    • createConfig

      public static long createConfig(String modelDir, String paramDir, ai.djl.Device device)

    • enableMKLDNN

      public static void enableMKLDNN(long config)

    • removePass

      public static void removePass(long config, String pass)

    • disableGLog

      public static void disableGLog(long config)

    • cpuMathLibraryNumThreads

      public static void cpuMathLibraryNumThreads(long config, int thread)

    • switchIrOptim

      public static void switchIrOptim(long config, boolean condition)

    • useFeedFetchOp

      public static void useFeedFetchOp(long config)

    • enableONNXRuntime

      public static void enableONNXRuntime(long config)

    • enableOrtOptimization

      public static void enableOrtOptimization(long config)

    • deleteConfig

      public static void deleteConfig(long config)

    • createPredictor

      public static long createPredictor(long config)

    • clonePredictor

      public static long clonePredictor(PaddlePredictor predictor)

    • deletePredictor

      public static void deletePredictor(PaddlePredictor predictor)

    • predictorForward

      public static ai.djl.ndarray.NDList predictorForward(PaddlePredictor predictor, PpNDArray[] inputs, String[] inputNames)

    • getInputNames

      public static String[] getInputNames(PaddlePredictor predictor)