File tree Expand file tree Collapse file tree 1 file changed +20
-1
lines changed Expand file tree Collapse file tree 1 file changed +20
-1
lines changed Original file line number Diff line number Diff 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 ;
You can’t perform that action at this time.
0 commit comments