Source code for webdnn.graph.graph

from typing import Iterable

from webdnn.graph.variable import Variable

WEBDNN_LICENSE = "(C) Machine Intelligence Laboratory (The University of Tokyo), MIT License"


class Graph:
    """Graph(inputs, outputs)

    Computation graph of DNN model.

    Args:
        inputs (list of :class:`~webdnn.Variable`): input variables
        outputs (list of :class:`~webdnn.Variable`): output variables
    """

    def __init__(self,
                 inputs: Iterable[Variable],
                 outputs: Iterable[Variable]):
        self.inputs = list(inputs)
        self.outputs = list(outputs)
        self.licenses = {"webdnn": WEBDNN_LICENSE}

    def __repr__(self):  # pragma: no coverage
        return f"""<{self.__class__.__name__} inputs={self.inputs}, outputs={self.outputs}>"""

    def __str__(self):  # pragma: no coverage
        return self.__repr__()