blob: b24d45d634f3aa1c0d732c40036fc34c85660f44 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
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)
|
