Remove usages of bridge fallback.

PiperOrigin-RevId: 622990929
tensorflow · Apr 9, 2024 · 98570a6 · 98570a6
1 parent f872181
commit 98570a6
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 8 deletions.
diff --git a/tensorflow_serving/servables/tensorflow/tfrt_saved_model_factory.cc b/tensorflow_serving/servables/tensorflow/tfrt_saved_model_factory.cc
@@ -206,9 +206,7 @@ absl::Status TfrtSavedModelFactory::CreateTfrtSavedModelWithMetadata(
         options.graph_execution_options.runtime_config));
   }
   if (config_.target_tpu()) {
-    compile_options.device_target = config_.enable_mlir_bridge_fallback()
-                                        ? TfrtDeviceInfraTarget::kBridgeFallback
-                                        : TfrtDeviceInfraTarget::kTpurt;
+    compile_options.device_target = TfrtDeviceInfraTarget::kTpurt;
   } else if (config_.enable_tfrt_gpu()) {
     compile_options.device_target = TfrtDeviceInfraTarget::kGpu;
   } else {

diff --git a/tensorflow_serving/servables/tensorflow/tfrt_saved_model_source_adapter.proto b/tensorflow_serving/servables/tensorflow/tfrt_saved_model_source_adapter.proto
@@ -6,7 +6,7 @@ import "tensorflow/core/protobuf/config.proto";
 import "tensorflow_serving/servables/tensorflow/session_bundle_config.proto";
 
 message TfrtSavedModelConfig {
-  reserved 1031;
+  reserved 1019, 1031;
   reserved "online_cost_analysis_normalize_ratio";
 
   // A SessionBundleConfig.
@@ -117,10 +117,6 @@ message TfrtSavedModelConfig {
   // If true, use fuesd op for tpu compile, execute and data transfer.
   bool use_fused_tpu_op = 1018;
 
-  // If true, TFRT will use fallback to handle unsupported features in MLIR
-  // Bridge.
-  bool enable_mlir_bridge_fallback = 1019;
-
   // If true, tf.While's iterations will be parallelized.
   bool enable_while_parallel_iterations = 1023;