aboutsummaryrefslogtreecommitdiff
path: root/vcpkg/ports/tensorflow-common/convert_lib_params_linux.py
diff options
context:
space:
mode:
Diffstat (limited to 'vcpkg/ports/tensorflow-common/convert_lib_params_linux.py')
-rw-r--r--vcpkg/ports/tensorflow-common/convert_lib_params_linux.py30
1 files changed, 30 insertions, 0 deletions
diff --git a/vcpkg/ports/tensorflow-common/convert_lib_params_linux.py b/vcpkg/ports/tensorflow-common/convert_lib_params_linux.py
new file mode 100644
index 0000000..b24d45d
--- /dev/null
+++ b/vcpkg/ports/tensorflow-common/convert_lib_params_linux.py
@@ -0,0 +1,30 @@
+import sys
+
+version = sys.argv[1]
+lib_suffix = "" if len(sys.argv) < 3 else sys.argv[2]
+
+with open(f"libtensorflow{lib_suffix}.a.{version}-2.params", "w") as f_out:
+ parts = []
+ with open(f"libtensorflow_framework.so.{version}-2.params", "r") as f_in:
+ skip_next = False
+ for line in f_in:
+ if skip_next:
+ skip_next = False
+ continue
+ if line.startswith("-o"):
+ skip_next = True
+ elif line.startswith("bazel-out"):
+ f_out.write(line)
+ parts.append(line)
+ parts = set(parts)
+ with open(f"libtensorflow{lib_suffix}.so.{version}-2.params", "r") as f_in:
+ skip_next = False
+ for line in f_in:
+ if skip_next:
+ skip_next = False
+ continue
+ if line.startswith("-o"):
+ skip_next = True
+ elif line.startswith("bazel-out"):
+ if line not in parts:
+ f_out.write(line)