Package ai.djl.tensorflow.engine.javacpp

Class JavacppUtils

java.lang.Object
ai.djl.tensorflow.engine.javacpp.JavacppUtils
public final class JavacppUtils extends Object
A class containing utilities to interact with the TensorFlow Engine's Javacpp layer.

  • Method Summary

    Modifier and Type
    Method
    Description
    static org.tensorflow.internal.c_api.TFE_Context
    createEagerSession(boolean async, int devicePlacementPolicy, org.tensorflow.proto.framework.ConfigProto config)
     
    static org.tensorflow.internal.c_api.TFE_TensorHandle
    createEmptyTFETensor(ai.djl.ndarray.types.Shape shape, ai.djl.ndarray.types.DataType dataType, org.tensorflow.internal.c_api.TFE_Context eagerSessionHandle, ai.djl.Device device)
     
    static ai.djl.util.Pair<org.tensorflow.internal.c_api.TF_Tensor,org.tensorflow.internal.c_api.TFE_TensorHandle>
    createStringTensor(long[] dims, ByteBuffer[] src)
     
    static org.tensorflow.internal.c_api.TFE_TensorHandle
    createTFETensor(org.tensorflow.internal.c_api.TF_Tensor handle)
     
    static org.tensorflow.internal.c_api.TFE_TensorHandle
    createTFETensorFromByteBuffer(ByteBuffer buf, ai.djl.ndarray.types.Shape shape, ai.djl.ndarray.types.DataType dataType, org.tensorflow.internal.c_api.TFE_Context eagerSessionHandle, ai.djl.Device device)
     
    static ai.djl.Device
    fromTfDevice(String device)
     
    static ByteBuffer
    getByteBuffer(org.tensorflow.internal.c_api.TFE_TensorHandle handle)
     
    static ai.djl.ndarray.types.DataType
    getDataType(org.tensorflow.internal.c_api.TFE_TensorHandle handle)
     
    static ai.djl.Device
    getDevice(org.tensorflow.internal.c_api.TFE_TensorHandle handle)
     
    static ai.djl.util.Pair<org.tensorflow.internal.c_api.TF_Operation,Integer>
    getGraphOperationByName(org.tensorflow.internal.c_api.TF_Graph graphHandle, String operation)
     
    static org.tensorflow.proto.framework.ConfigProto
    getSessionConfig()
     
    static ai.djl.ndarray.types.Shape
    getShape(org.tensorflow.internal.c_api.TFE_TensorHandle handle)
     
    static String[]
    getString(org.tensorflow.internal.c_api.TFE_TensorHandle handle, int count, Charset charset)
     
    static SavedModelBundle
    loadSavedModelBundle(String exportDir, String[] tags, org.tensorflow.proto.framework.ConfigProto config, org.tensorflow.proto.framework.RunOptions runOptions)
     
    static org.tensorflow.internal.c_api.TF_Tensor
    resolveTFETensor(org.tensorflow.internal.c_api.TFE_TensorHandle handle)
     
    static org.tensorflow.internal.c_api.TF_Tensor[]
    runSession(org.tensorflow.internal.c_api.TF_Session handle, org.tensorflow.proto.framework.RunOptions runOptions, org.tensorflow.internal.c_api.TF_Tensor[] inputTensorHandles, org.tensorflow.internal.c_api.TF_Operation[] inputOpHandles, int[] inputOpIndices, org.tensorflow.internal.c_api.TF_Operation[] outputOpHandles, int[] outputOpIndices, org.tensorflow.internal.c_api.TF_Operation[] targetOpHandles)
     
    static void
    setByteBuffer(org.tensorflow.internal.c_api.TFE_TensorHandle handle, ByteBuffer data)
     
    static org.tensorflow.internal.c_api.TFE_TensorHandle
    toDevice(org.tensorflow.internal.c_api.TFE_TensorHandle handle, org.tensorflow.internal.c_api.TFE_Context eagerSessionHandle, ai.djl.Device device)
     
    static String
    toTfDevice(ai.djl.Device device)
     

    Methods inherited from class java.lang.Object

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

  • Method Details

    • loadSavedModelBundle

      public static SavedModelBundle loadSavedModelBundle(String exportDir, String[] tags, org.tensorflow.proto.framework.ConfigProto config, org.tensorflow.proto.framework.RunOptions runOptions)

    • getGraphOperationByName

      public static ai.djl.util.Pair<org.tensorflow.internal.c_api.TF_Operation,Integer> getGraphOperationByName(org.tensorflow.internal.c_api.TF_Graph graphHandle, String operation)

    • runSession

      public static org.tensorflow.internal.c_api.TF_Tensor[] runSession(org.tensorflow.internal.c_api.TF_Session handle, org.tensorflow.proto.framework.RunOptions runOptions, org.tensorflow.internal.c_api.TF_Tensor[] inputTensorHandles, org.tensorflow.internal.c_api.TF_Operation[] inputOpHandles, int[] inputOpIndices, org.tensorflow.internal.c_api.TF_Operation[] outputOpHandles, int[] outputOpIndices, org.tensorflow.internal.c_api.TF_Operation[] targetOpHandles)

    • createEagerSession

      public static org.tensorflow.internal.c_api.TFE_Context createEagerSession(boolean async, int devicePlacementPolicy, org.tensorflow.proto.framework.ConfigProto config)

    • getDevice

      public static ai.djl.Device getDevice(org.tensorflow.internal.c_api.TFE_TensorHandle handle)

    • getDataType

      public static ai.djl.ndarray.types.DataType getDataType(org.tensorflow.internal.c_api.TFE_TensorHandle handle)

    • getShape

      public static ai.djl.ndarray.types.Shape getShape(org.tensorflow.internal.c_api.TFE_TensorHandle handle)

    • createEmptyTFETensor

      public static org.tensorflow.internal.c_api.TFE_TensorHandle createEmptyTFETensor(ai.djl.ndarray.types.Shape shape, ai.djl.ndarray.types.DataType dataType, org.tensorflow.internal.c_api.TFE_Context eagerSessionHandle, ai.djl.Device device)

    • createStringTensor

      public static ai.djl.util.Pair<org.tensorflow.internal.c_api.TF_Tensor,org.tensorflow.internal.c_api.TFE_TensorHandle> createStringTensor(long[] dims, ByteBuffer[] src)

    • createTFETensorFromByteBuffer

      public static org.tensorflow.internal.c_api.TFE_TensorHandle createTFETensorFromByteBuffer(ByteBuffer buf, ai.djl.ndarray.types.Shape shape, ai.djl.ndarray.types.DataType dataType, org.tensorflow.internal.c_api.TFE_Context eagerSessionHandle, ai.djl.Device device)

    • resolveTFETensor

      public static org.tensorflow.internal.c_api.TF_Tensor resolveTFETensor(org.tensorflow.internal.c_api.TFE_TensorHandle handle)

    • createTFETensor

      public static org.tensorflow.internal.c_api.TFE_TensorHandle createTFETensor(org.tensorflow.internal.c_api.TF_Tensor handle)

    • getString

      public static String[] getString(org.tensorflow.internal.c_api.TFE_TensorHandle handle, int count, Charset charset)

    • setByteBuffer

      public static void setByteBuffer(org.tensorflow.internal.c_api.TFE_TensorHandle handle, ByteBuffer data)

    • getByteBuffer

      public static ByteBuffer getByteBuffer(org.tensorflow.internal.c_api.TFE_TensorHandle handle)

    • toDevice

      public static org.tensorflow.internal.c_api.TFE_TensorHandle toDevice(org.tensorflow.internal.c_api.TFE_TensorHandle handle, org.tensorflow.internal.c_api.TFE_Context eagerSessionHandle, ai.djl.Device device)

    • getSessionConfig

      public static org.tensorflow.proto.framework.ConfigProto getSessionConfig()

    • fromTfDevice

      public static ai.djl.Device fromTfDevice(String device)

    • toTfDevice

      public static String toTfDevice(ai.djl.Device device)