# See the License for the specific language governing permissions and
# limitations under the License.
# ==============================================================================

# RUN: %p/control_flow_upgrade_legacy_v1 | FileCheck %s

# pylint: disable=missing-docstring,line-too-long
import tensorflow.compat.v1 as tf
from tensorflow.compiler.mlir.tensorflow.tests.tf_saved_model import common_v1

    srcs = ["shapes_for_arguments.py"],
    deps = [
        ":common",
        "//tensorflow:tensorflow_py",
    ],
)

py_strict_binary(
    name = "control_flow_upgrade_legacy_v1",
    srcs = ["control_flow_upgrade_legacy_v1.py"],
    deps = [
        ":common_v1",
        "//tensorflow:tensorflow_py",
        "//tensorflow/python/ops:control_flow_ops",
    ],
)

py_strict_binary(