Ejemplos de get_output_metadata en Python

Ejemplo n.º 1

Archivo: loader.py Proyecto: stjordanis/TensorRT

    def call_impl(self):
        """
        Returns:
            Tuple[tf.Graph, Sequence[str]]: The TensorFlow graph, and the names of its outputs.
        """
        (graph, outputs), _ = util.invoke_if_callable(self._graph)

        if self.outputs == constants.MARK_ALL:
            outputs = list(
                tf_util.get_output_metadata(graph, layerwise=True).keys())
        elif self.outputs is not None:
            outputs = self.outputs

        return graph, outputs

Ejemplo n.º 2

Archivo: loader.py Proyecto: celidos/TensorRT_study

    def __call__(self):
        """
        Modifies a TensorFlow graph.

        Returns:
            Tuple[tf.Graph, Sequence[str]]: The TensorFlow graph, and the names of its outputs.
        """
        (graph, outputs), _ = misc.try_call(self._graph)

        if self.outputs == constants.MARK_ALL:
            outputs = list(
                tf_util.get_output_metadata(graph, layerwise=True).keys())
        elif self.outputs is not None:
            outputs = self.outputs

        return graph, outputs