Python get_output_metadataの例

コード例 #1

ファイル: loader.py プロジェクト: 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

コード例 #2

ファイル: loader.py プロジェクト: 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