From b09c0e9938f1cc0c9542afd9ec7dbe5d506bf2db Mon Sep 17 00:00:00 2001 From: Sebastian Schmidt Date: Mon, 17 Oct 2022 15:54:36 -0700 Subject: [PATCH] Internal change PiperOrigin-RevId: 481761007 --- mediapipe/calculators/tflite/BUILD | 1 + mediapipe/calculators/tflite/tflite_model_calculator.cc | 6 ++++++ 2 files changed, 7 insertions(+) diff --git a/mediapipe/calculators/tflite/BUILD b/mediapipe/calculators/tflite/BUILD index 1d2f279a..2007a4fe 100644 --- a/mediapipe/calculators/tflite/BUILD +++ b/mediapipe/calculators/tflite/BUILD @@ -331,6 +331,7 @@ cc_library( "//mediapipe/framework:calculator_framework", "//mediapipe/framework:packet", "//mediapipe/framework/port:ret_check", + "@com_google_absl//absl/status", "@org_tensorflow//tensorflow/lite:framework", ], alwayslink = 1, diff --git a/mediapipe/calculators/tflite/tflite_model_calculator.cc b/mediapipe/calculators/tflite/tflite_model_calculator.cc index d118e878..c347d07f 100644 --- a/mediapipe/calculators/tflite/tflite_model_calculator.cc +++ b/mediapipe/calculators/tflite/tflite_model_calculator.cc @@ -16,6 +16,7 @@ #include #include +#include "absl/status/status.h" #include "mediapipe/framework/calculator_framework.h" #include "mediapipe/framework/packet.h" #include "mediapipe/framework/port/ret_check.h" @@ -81,6 +82,7 @@ class TfLiteModelCalculator : public CalculatorBase { } if (cc->InputSidePackets().HasTag("MODEL_FD")) { +#ifdef ABSL_HAVE_MMAP model_packet = cc->InputSidePackets().Tag("MODEL_FD"); const auto& model_fd = model_packet.Get>(); @@ -89,6 +91,10 @@ class TfLiteModelCalculator : public CalculatorBase { tflite::DefaultErrorReporter()); model = tflite::FlatBufferModel::BuildFromAllocation( std::move(model_allocation), tflite::DefaultErrorReporter()); +#elif + return absl::FailedPreconditionError( + "Loading by file descriptor is not supported on this platform.") +#endif } RET_CHECK(model) << "Failed to load TfLite model from blob.";