Location via proxy:   [ UP ]  
[Report a bug]   [Manage cookies]                
Skip to content

Commit a018b73

Browse files
committed
Updates proto-update script to match new reference to 'tsl' dir
(cherry picked from commit e39f201)
1 parent 4bed196 commit a018b73

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

tensorboard/compat/proto/proto_test.py

+1-1
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-
("tsl/protobuf", "tensorboard/compat/proto"),
158+
("xla/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

+3-3
Original file line numberDiff line numberDiff line change
@@ -28,8 +28,8 @@ rsync --existing "$1"/tensorflow/core/util/*.proto tensorboard/compat/proto/
2828
rsync --existing "$1"/tensorflow/python/framework/*.proto tensorboard/compat/proto/
2929
# The "TSL" protos are now in their own repo, but they are included as a
3030
# "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/
31+
# the xla/tsl/protobuf path that is replaced by the command below.
32+
rsync --existing "$1"/third_party/xla/xla/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|tsl/protobuf|tensorboard/compat/proto|g;' \
43+
-e 's|xla/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)