diff options
| -rw-r--r-- | verif/frameworks/test_builder.py | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/verif/frameworks/test_builder.py b/verif/frameworks/test_builder.py index 4bf8070..900f5fa 100644 --- a/verif/frameworks/test_builder.py +++ b/verif/frameworks/test_builder.py @@ -1,9 +1,13 @@ # Copyright (c) 2020-2024, ARM Limited. # SPDX-License-Identifier: Apache-2.0 +import os + import numpy as np import tensorflow as tf from frameworks.tensor_gen import TGen +os.environ["TF_USE_LEGACY_KERAS"] = "1" + class TBuilder: """The member functions build the tensorflow operators into small networks |