<filename>model_compiler/src/model_compiler/models/irs/tf_frozen_graph_model.py
# Copyright 2019 ZTE corporation. All Rights Reserved.
# SPDX-License-Identifier: Apache-2.0

from typing import NamedTuple, Optional, Sequence

import tensorflow as tf

from ..data_format import DataFormat


class Input(NamedTuple):
    name: str
    data_format: Optional[DataFormat] = None


class TensorFlowFrozenGraphModel(NamedTuple):
    graph_def: tf.compat.v1.GraphDef
    inputs: Sequence[Input]
    outputs: Sequence[str]

<filename>model_compiler/src/model_compiler/models/irs/tf_frozen_graph_model.py
# Copyright 2019 ZTE corporation. All Rights Reserved.
# SPDX-License-Identifier: Apache-2.0

from typing import NamedTuple, Optional, Sequence

import tensorflow as tf

from ..data_format import DataFormat


class Input(NamedTuple):
    name: str
    data_format: Optional[DataFormat] = None


class TensorFlowFrozenGraphModel(NamedTuple):
    graph_def: tf.compat.v1.GraphDef
    inputs: Sequence[Input]
    outputs: Sequence[str]

# Copyright 2019 ZTE corporation. All Rights Reserved. # SPDX-License-Identifier: Apache-2.0