Package ai.djl.tensorflow.engine

Class SavedModelBundle

  • All Implemented Interfaces:
    java.lang.AutoCloseable
    public class SavedModelBundle
extends java.lang.Object
implements java.lang.AutoCloseable
    The wrapper class for native resources required for SavedModelBundle.

    • Constructor Summary

      Constructors 
      Constructor Description
      SavedModelBundle​(org.tensorflow.internal.c_api.TF_Graph graphHandle, org.tensorflow.internal.c_api.TF_Session sessionHandle, org.tensorflow.proto.framework.MetaGraphDef metaGraphDef)  

    • Method Summary

      All Methods Instance Methods Concrete Methods 
      Modifier and Type Method Description
      void close()
      org.tensorflow.internal.c_api.TF_Graph getGraph()
      Returns the graph handle.
      org.tensorflow.proto.framework.MetaGraphDef getMetaGraphDef()
      Returns the MetaGraphDef protol buf.
      org.tensorflow.internal.c_api.TF_Session getSession()
      Returns the session handle.

      • Methods inherited from class java.lang.Object

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

    • Constructor Detail

      • SavedModelBundle

        public SavedModelBundle​(org.tensorflow.internal.c_api.TF_Graph graphHandle,
                        org.tensorflow.internal.c_api.TF_Session sessionHandle,
                        org.tensorflow.proto.framework.MetaGraphDef metaGraphDef)

    • Method Detail

      • getGraph

        public org.tensorflow.internal.c_api.TF_Graph getGraph()
        Returns the graph handle.
        Returns:
        the graph handle

      • getSession

        public org.tensorflow.internal.c_api.TF_Session getSession()
        Returns the session handle.
        Returns:
        the session handle

      • getMetaGraphDef

        public org.tensorflow.proto.framework.MetaGraphDef getMetaGraphDef()
        Returns the MetaGraphDef protol buf.
        Returns:
        the MetaGraphDef protol buf

      • close

        public void close()
        Specified by:
        close in interface java.lang.AutoCloseable