No description

PiperOrigin-RevId: 481235204
twtstudio · Oct 14, 2022 · d7a4680 · d7a4680
1 parent 6642018
commit d7a4680
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/lite/examples/classification_by_retrieval/lib/model_builder.cc b/lite/examples/classification_by_retrieval/lib/model_builder.cc
@@ -1,5 +1,7 @@
 #include "lib/model_builder.h"
 
+#include <memory>
+
 #include "absl/status/status.h"
 #include "absl/strings/str_format.h"
 #include "absl/strings/str_join.h"
@@ -61,8 +63,8 @@ ModelBuilder::CreateFromImageEmbedderOptions(
     return absl::InvalidArgumentError("Failed to build model from file: " +
                                       model_file);
   }
-  return absl::make_unique<ModelBuilder>(std::move(image_embedder),
-                                         std::move(model));
+  return std::make_unique<ModelBuilder>(std::move(image_embedder),
+                                        std::move(model));
 }
 
 void ModelBuilder::SetMetadata(

diff --git a/lite/examples/classification_by_retrieval/lib/model_builder.h b/lite/examples/classification_by_retrieval/lib/model_builder.h
@@ -43,7 +43,7 @@ class ModelBuilder {
       std::unique_ptr<::tflite::task::vision::ImageEmbedder> image_embedder,
       std::unique_ptr<::tflite::FlatBufferModel> model,
       std::unique_ptr<TfLiteCbRBuilder> tflite_cbr_builder =
-          absl::make_unique<TfLiteCbRBuilder>());
+          std::make_unique<TfLiteCbRBuilder>());
 
   // Initializes the ModelBuilder from the provided ImageEmbedderOptions.
   static tflite::support::StatusOr<std::unique_ptr<ModelBuilder>>