Skip to content

Commit dcb1bb6

Browse files
authored
Updates compat protos with latest snapshot from TF (#6744)
Just updating some proto definitions for compatibility with no-TF mode.
1 parent 7c292a5 commit dcb1bb6

File tree

1 file changed

+20
-1
lines changed

1 file changed

+20
-1
lines changed

tensorboard/compat/proto/config.proto

+20-1
Original file line numberDiff line numberDiff line change
@@ -673,6 +673,25 @@ message ConfigProto {
673673
// Whether runtime execution uses TFRT.
674674
bool use_tfrt = 18;
675675

676+
// If true, use Pathways with TFRT API for multi host support.
677+
bool enable_multi_host = 27;
678+
679+
// Port for the Pathways server. Ignored if enable_multi_host=false.
680+
int32 backend_server_port = 28;
681+
682+
// If true, TFRT will use TPU specific compiler passes and perform TPU
683+
// specific initialization.
684+
bool target_tpu = 29;
685+
686+
// If true, TFRT will use GPU specific compiler passes and perform GPU
687+
// specific initialization.
688+
bool target_gpu = 30;
689+
690+
// The threshold to merge small streams in TFRT. The stream with cost
691+
// smaller than the threshold will be merged. Setting it to value 1
692+
// disables all merges.
693+
int32 stream_merge_threshold = 31;
694+
676695
// The field "coordination_service was previously specified as a string;
677696
// this has been replaced with a message below.
678697
reserved 19;
@@ -711,7 +730,7 @@ message ConfigProto {
711730

712731
reserved 25;
713732

714-
// Next: 27
733+
// Next: 32
715734
}
716735

717736
Experimental experimental = 16;

0 commit comments

Comments
 (0)