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 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	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 ai.djl.ndarray.types.Shape	getShapeFromNd(PpNDArray array)
static java.lang.String	getVersion()
static PpNDArray[]	predictorForward(PaddlePredictor predictor, PpNDArray[] inputs, java.lang.String[] inputNames)
static void	setNdName(PpNDArray array, java.lang.String name)
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)

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)

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 PpNDArray[] 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()