ai.djl.paddlepaddle.jni

Class JniUtils

java.lang.Object

ai.djl.paddlepaddle.jni.JniUtils

public final class JniUtils
extends java.lang.Object

A class containing utilities to interact with the Paddle Engine's Java Native Interface (JNI) layer.

Method Summary

Modifier and Type	Method and Description
static long	clonePredictor(PaddlePredictor predictor)
static void	cpuMathLibraryNumThreads(long config, int thread)
static long	createConfig(java.lang.String modelDir, java.lang.String paramDir, ai.djl.Device device)
static PpNDArray	createNdArray(PpNDManager manager, java.nio.ByteBuffer data, ai.djl.ndarray.types.Shape shape, ai.djl.ndarray.types.DataType dtype)
static long	createPredictor(long config)
static void	deleteConfig(long config)
static void	deleteNd(java.lang.Long handle)
static void	deletePredictor(PaddlePredictor predictor)
static void	disableGLog(long config)
static void	enableMKLDNN(long config)
static java.nio.ByteBuffer	getByteBufferFromNd(PpNDArray array)
static ai.djl.ndarray.types.DataType	getDTypeFromNd(PpNDArray array)
static java.lang.String[]	getInputNames(PaddlePredictor predictor)
static java.lang.String	getNameFromNd(PpNDArray array)
static long[][]	getNdLoD(PpNDArray array)
static ai.djl.ndarray.types.Shape	getShapeFromNd(PpNDArray array)
static java.lang.String	getVersion()
static ai.djl.ndarray.NDList	predictorForward(PaddlePredictor predictor, PpNDArray[] inputs, java.lang.String[] inputNames)
static void	removePass(long config, java.lang.String pass)
static void	setNdLoD(PpNDArray array, long[][] lod)
static void	setNdName(PpNDArray array, java.lang.String name)
static void	switchIrOptim(long config, boolean condition)
static void	useFeedFetchOp(long config)

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Method Detail

createNdArray

public static PpNDArray createNdArray(PpNDManager manager,
                                      java.nio.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 java.nio.ByteBuffer getByteBufferFromNd(PpNDArray array)

getShapeFromNd

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

setNdName

public static void setNdName(PpNDArray array,
                             java.lang.String name)

getNameFromNd

public static java.lang.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(java.lang.Long handle)

createConfig

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

enableMKLDNN

public static void enableMKLDNN(long config)

removePass

public static void removePass(long config,
                              java.lang.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)

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,
                                                     java.lang.String[] inputNames)

getInputNames

public static java.lang.String[] getInputNames(PaddlePredictor predictor)

getVersion

public static java.lang.String getVersion()