Package ai.djl.tensorflow.engine

Class SavedModelBundle

java.lang.Object
ai.djl.tensorflow.engine.SavedModelBundle
All Implemented Interfaces:
AutoCloseable
public class SavedModelBundle extends Object implements 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

    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 Details

    • 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 Details

    • 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 AutoCloseable