Package ai.djl.tensorflow.engine.javacpp

Class JavacppUtils

  • public final class JavacppUtils
extends java.lang.Object
    A class containing utilities to interact with the TensorFlow Engine's Javacpp layer.

    • Method Summary

      All Methods Static Methods Concrete Methods 
      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, java.nio.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​(java.nio.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​(java.lang.String device)  
      static java.nio.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,​java.lang.Integer> getGraphOperationByName​(org.tensorflow.internal.c_api.TF_Graph graphHandle, java.lang.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 java.lang.String[] getString​(org.tensorflow.internal.c_api.TFE_TensorHandle handle, int count, java.nio.charset.Charset charset)  
      static SavedModelBundle loadSavedModelBundle​(java.lang.String exportDir, java.lang.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, java.nio.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 java.lang.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 Detail

      • loadSavedModelBundle

        public static SavedModelBundle loadSavedModelBundle​(java.lang.String exportDir,
                                                    java.lang.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,​java.lang.Integer> getGraphOperationByName​(org.tensorflow.internal.c_api.TF_Graph graphHandle,
                                                                                                                           java.lang.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,
                                                                                                                                                java.nio.ByteBuffer[] src)

      • createTFETensorFromByteBuffer

        public static org.tensorflow.internal.c_api.TFE_TensorHandle createTFETensorFromByteBuffer​(java.nio.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 java.lang.String[] getString​(org.tensorflow.internal.c_api.TFE_TensorHandle handle,
                                           int count,
                                           java.nio.charset.Charset charset)

      • setByteBuffer

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

      • getByteBuffer

        public static java.nio.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​(java.lang.String device)

      • toTfDevice

        public static java.lang.String toTfDevice​(ai.djl.Device device)