Exported FindMetaGraphDef function.

PiperOrigin-RevId: 609799877
tensorflow · Feb 23, 2024 · 0df0975 · 0df0975
1 parent 5d803d4
commit 0df0975
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tensorflow_serving/servables/tensorflow/tfrt_saved_model_factory.cc b/tensorflow_serving/servables/tensorflow/tfrt_saved_model_factory.cc
@@ -182,6 +182,8 @@ absl::Status TfrtSavedModelFactory::CreateTfrtSavedModelWithMetadata(
 
   TF_RETURN_IF_ERROR(RegisterCustomBackend(options.graph_execution_options));
 
+  // TODO(b/326069213): Consider using arena allocation when loading a
+  // MetaGraphDef.
   tensorflow::MetaGraphDef meta_graph_def;
   TF_RETURN_IF_ERROR(tensorflow::ReadMetaGraphDefFromSavedModel(
       std::string(path), saved_model_tags, &meta_graph_def));