Fixed proto2 to google::protobuf conversion.

PiperOrigin-RevId: 628087579
tensorflow · Apr 25, 2024 · 47d4414 · 47d4414
1 parent 1cd945a
commit 47d4414
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/tensorflow/core/tfrt/saved_model/saved_model_util.cc b/tensorflow/core/tfrt/saved_model/saved_model_util.cc
@@ -367,7 +367,7 @@ absl::Status DeserializeAoTMlirModule(
 }
 
 CallableOptions CombineSignatureDefs(
-    const proto2::Map<std::string, SignatureDef>& signature_defs) {
+    const google::protobuf::Map<std::string, SignatureDef>& signature_defs) {
   CallableOptions callable_options;
   for (const auto& sig_iter : signature_defs) {
     const auto& signature_def = sig_iter.second;

diff --git a/tensorflow/core/tfrt/saved_model/saved_model_util.h b/tensorflow/core/tfrt/saved_model/saved_model_util.h
@@ -142,7 +142,7 @@ absl::Status DeserializeAoTMlirModule(
     mlir::OwningOpRef<mlir::ModuleOp>* mlir_module);
 
 CallableOptions CombineSignatureDefs(
-    const proto2::Map<std::string, SignatureDef>& signature_defs);
+    const google::protobuf::Map<std::string, SignatureDef>& signature_defs);
 
 void RegisterTfrtDialectsForAot(mlir::DialectRegistry& registry);