Skip to content

Commit c97601a

Browse files
committed
Updates script to sync TF compat protos into TB repo.
1 parent 143508b commit c97601a

File tree

2 files changed

+6
-6
lines changed

2 files changed

+6
-6
lines changed

tensorboard/compat/proto/proto_test.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -155,7 +155,7 @@
155155
("tensorflow/core/protobuf/", "tensorboard/compat/proto/"),
156156
("tensorflow/core/util/", "tensorboard/compat/proto/"),
157157
("tensorflow/python/framework/", "tensorboard/compat/proto/"),
158-
("tensorflow/tsl/protobuf", "tensorboard/compat/proto"),
158+
("tsl/protobuf", "tensorboard/compat/proto"),
159159
('package: "tensorflow.tfprof"', 'package: "tensorboard"'),
160160
('package: "tensorflow"', 'package: "tensorboard"'),
161161
('type_name: ".tensorflow.tfprof', 'type_name: ".tensorboard'),

tensorboard/compat/proto/update.sh

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -26,10 +26,10 @@ rsync --existing "$1"/tensorflow/core/protobuf/*.proto tensorboard/compat/proto/
2626
rsync --existing "$1"/tensorflow/core/profiler/*.proto tensorboard/compat/proto/
2727
rsync --existing "$1"/tensorflow/core/util/*.proto tensorboard/compat/proto/
2828
rsync --existing "$1"/tensorflow/python/framework/*.proto tensorboard/compat/proto/
29-
if [ -d $1"/tensorflow/tsl/protobuf" ] # This directory first appears in TF 2.11
30-
then
31-
rsync --existing "$1"/tensorflow/tsl/protobuf/*.proto tensorboard/compat/proto/
32-
fi
29+
# The "TSL" protos are now in their own repo, but they are included as a
30+
# "vendored" package within TF. This dir contains protos that are imported with
31+
# the tsl/protobuf path that is replaced by the command below.
32+
rsync --existing "$1"/third_party/xla/third_party/tsl/tsl/protobuf/*.proto tensorboard/compat/proto/
3333

3434
# Rewrite file paths and package names and disable LINT checks.
3535
#
@@ -40,7 +40,7 @@ find tensorboard/compat/proto/ -type f -name '*.proto' -exec perl -pi \
4040
-e 's|tensorflow/core/profiler|tensorboard/compat/proto|g;' \
4141
-e 's|tensorflow/core/util|tensorboard/compat/proto|g;' \
4242
-e 's|tensorflow/python/framework|tensorboard/compat/proto|g;' \
43-
-e 's|tensorflow/tsl/protobuf|tensorboard/compat/proto|g;' \
43+
-e 's|tsl/protobuf|tensorboard/compat/proto|g;' \
4444
-e 's|package tensorflow.tfprof;|package tensorboard;|g;' \
4545
-e 's|package tensorflow;|package tensorboard;|g;' \
4646
-e 's|tensorflow\.DataType|tensorboard.DataType|g;' \

0 commit comments

Comments
 (0)