aboutsummaryrefslogtreecommitdiffhomepage
path: root/configure.py
diff options
context:
space:
mode:
authorGravatar Anna R <annarev@google.com>2018-09-14 12:44:31 -0700
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2018-09-14 12:50:07 -0700
commita9a1d5a673ad085777e6a8b14cbe39a427493e51 (patch)
treef4a7cdb32238367a9036074aeb722f1a7a105bce /configure.py
parent9da83f0701bcece95372ee8da09f886dfd2fa2a1 (diff)
Add --config=v2 option to the .bazelrc file.
PiperOrigin-RevId: 213027176
Diffstat (limited to 'configure.py')
-rw-r--r--configure.py3
1 files changed, 3 insertions, 0 deletions
diff --git a/configure.py b/configure.py
index 52a513779e..e9d162fbd2 100644
--- a/configure.py
+++ b/configure.py
@@ -1572,6 +1572,9 @@ def main():
if is_windows():
set_windows_build_flags(environ_cp)
+ # Add a config option to build TensorFlow 2.0 API.
+ write_to_bazelrc('build:v2 --define=tf_api_version=2')
+
if get_var(
environ_cp, 'TF_SET_ANDROID_WORKSPACE', 'android workspace',
False,