Add core selector support for TFRT+IFRT serving on tensorflow serving

PiperOrigin-RevId: 633836270
tensorflow · May 15, 2024 · 84a71a4 · 84a71a4
1 parent a2d636d
commit 84a71a4
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/tensorflow_serving/servables/tensorflow/tfrt_saved_model_source_adapter.proto b/tensorflow_serving/servables/tensorflow/tfrt_saved_model_source_adapter.proto
@@ -200,6 +200,10 @@ message TfrtSavedModelConfig {
 
   // Whether to use IFRT as device driver for device execution.
   bool tfrt_use_ifrt = 2020;
+
+  // Whether to enable core selector for TFRT+IFRT. It is only effective when
+  // tfrt_use_ifrt is true.
+  bool enable_tfrt_use_ifrt_core_selector = 2021;
 }
 
 // Config proto for TfrtSavedModelSourceAdapter.