aboutsummaryrefslogtreecommitdiffstats
path: root/classes/tensorflow_ver.bbclass
diff options
context:
space:
mode:
Diffstat (limited to 'classes/tensorflow_ver.bbclass')
-rw-r--r--classes/tensorflow_ver.bbclass5
1 files changed, 5 insertions, 0 deletions
diff --git a/classes/tensorflow_ver.bbclass b/classes/tensorflow_ver.bbclass
new file mode 100644
index 0000000..c20602a
--- /dev/null
+++ b/classes/tensorflow_ver.bbclass
@@ -0,0 +1,5 @@
+# Build tensorflow 2.x by default
+TF_ARGS_EXTRA ??= "--config=v2"
+
+# Uncomment the following line to build tensorflow 1.x
+#TF_ARGS_EXTRA = "--config=v1"