Python load_graph Examples

Example #1

File: loader.py Project: stjordanis/TensorRT

 def call_impl(self):
     """
     Returns:
         Tuple[tf.Graph, Sequence[str]]: The TensorFlow graph, and the names of its outputs.
     """
     graph = tf_util.load_graph(self.path)
     return graph, tf_util.get_graph_output_names(graph)

Example #2

File: loader.py Project: celidos/TensorRT_study

    def __call__(self):
        """
        Loads a TensorFlow frozen model.

        Returns:
            Tuple[tf.Graph, Sequence[str]]: The TensorFlow graph, and the names of its outputs.
        """
        graph = tf_util.load_graph(self.path)
        return graph, tf_util.get_graph_output_names(graph)