Skip to content

Commit 6a3f1c9

Browse files
authored
Updates TF compat protos (tensorflow#7136)
Routine compat proto update before producing the next release that tracks the TensorFlow release.
1 parent 73820fa commit 6a3f1c9

5 files changed

Lines changed: 30 additions & 16 deletions

File tree

tensorboard/compat/proto/BUILD

Lines changed: 11 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -126,11 +126,6 @@ tb_proto_library(
126126
],
127127
)
128128

129-
tb_proto_library(
130-
name = "graph_debug_info",
131-
srcs = ["graph_debug_info.proto"],
132-
)
133-
134129
tb_proto_library(
135130
name = "graph",
136131
srcs = ["graph.proto"],
@@ -142,6 +137,11 @@ tb_proto_library(
142137
],
143138
)
144139

140+
tb_proto_library(
141+
name = "graph_debug_info",
142+
srcs = ["graph_debug_info.proto"],
143+
)
144+
145145
tb_proto_library(
146146
name = "histogram",
147147
srcs = ["histogram.proto"],
@@ -255,20 +255,20 @@ tb_proto_library(
255255
)
256256

257257
tb_proto_library(
258-
name = "tensor_description",
259-
srcs = ["tensor_description.proto"],
258+
name = "tensor",
259+
srcs = ["tensor.proto"],
260260
deps = [
261-
":allocation_description",
261+
":resource_handle",
262262
":tensor_shape",
263263
":types",
264264
],
265265
)
266266

267267
tb_proto_library(
268-
name = "tensor",
269-
srcs = ["tensor.proto"],
268+
name = "tensor_description",
269+
srcs = ["tensor_description.proto"],
270270
deps = [
271-
":resource_handle",
271+
":allocation_description",
272272
":tensor_shape",
273273
":types",
274274
],

tensorboard/compat/proto/config.proto

Lines changed: 13 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -772,6 +772,8 @@ message ConfigProto {
772772
// aims to negate its value.
773773
bool disable_optimize_for_static_graph = 24;
774774

775+
reserved 25;
776+
775777
// Whether eager remote execution will stream all the function calls or
776778
// allow them to happen in parallel. When true, streaming execution is
777779
// disabled, and parallel execution is allowed.
@@ -785,9 +787,18 @@ message ConfigProto {
785787
// is finalized.
786788
bool finalize_resource_manager = 34;
787789

788-
reserved 25;
790+
// If set, TF2XLA will dump compilation call arguments to the specified
791+
// directory.
792+
string tf2xla_dump_dir = 35;
793+
794+
// When set to true, the online op cost analysis will be enabled. This
795+
// will run the cost analysis once for the first model execution and
796+
// use the obtained cost to optimize subsequent executions.
797+
// Typically `stream_merge_threshold` should be tuned to set to a non-zero
798+
// value when this option is enabled.
799+
bool online_cost_analysis = 36;
789800

790-
// Next: 35
801+
// Next: 37
791802
}
792803

793804
Experimental experimental = 16;

tensorboard/compat/proto/types.proto

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -49,11 +49,11 @@ enum DataType {
4949
// bias, finite-only, with NaNs.
5050
DT_FLOAT8_E5M2FNUZ = 28; // 5 exponent bits, 2 mantissa bits, finite-only,
5151
// with NaN.
52-
5352
DT_INT4 = 29;
5453
DT_UINT4 = 30;
5554
DT_INT2 = 31;
5655
DT_UINT2 = 32;
56+
DT_FLOAT4_E2M1FN = 33; // 2 exponent bits, 1 mantissa bit, finite-only
5757

5858
// Do not use! These are only for TF1's obsolete reference Variables.
5959
// Every enum above should have a corresponding value below (verified by
@@ -91,6 +91,7 @@ enum DataType {
9191
DT_UINT4_REF = 130;
9292
DT_INT2_REF = 131;
9393
DT_UINT2_REF = 132;
94+
DT_FLOAT4_E2M1FN_REF = 133;
9495
}
9596
// DISABLED.ThenChange(
9697
// https://www.tensorflow.org/code/tensorflow/c/tf_datatype.h,
179 Bytes
Binary file not shown.

tensorboard/data/server/tensorboard.pb.rs

Lines changed: 4 additions & 2 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

0 commit comments

Comments
 (0)