Project import generated by Copybara.

GitOrigin-RevId: ec25bf2e416c3689477e82946fb69de2e53b9161

mediapipe/util/tflite/tflite_gpu_runner.cc

@@ -85,7 +85,7 @@ ObjectDef GetSSBOObjectDef(int channels) {
 
 absl::Status TFLiteGPURunner::InitializeWithModel(
     const tflite::FlatBufferModel& flatbuffer,
-    const tflite::OpResolver& op_resolver) {
+    const tflite::OpResolver& op_resolver, bool allow_quant_ops) {
   // GraphFloat32 is created twice because, when OpenCL and OpenGL backends are
   // initialized, different backend-specific graph transformations happen
   // in-place. As GraphFloat32 is not copyable by design, we keep two copies of
@@ -94,10 +94,10 @@ absl::Status TFLiteGPURunner::InitializeWithModel(
   // in the end of the initialization stage.
   graph_gl_ = std::make_unique<GraphFloat32>();
   graph_cl_ = std::make_unique<GraphFloat32>();
-  MP_RETURN_IF_ERROR(
-      BuildFromFlatBuffer(flatbuffer, op_resolver, graph_gl_.get()));
-  MP_RETURN_IF_ERROR(
-      BuildFromFlatBuffer(flatbuffer, op_resolver, graph_cl_.get()));
+  MP_RETURN_IF_ERROR(BuildFromFlatBuffer(flatbuffer, op_resolver,
+                                         graph_gl_.get(), allow_quant_ops));
+  MP_RETURN_IF_ERROR(BuildFromFlatBuffer(flatbuffer, op_resolver,
+                                         graph_cl_.get(), allow_quant_ops));
 
   for (const auto& input : graph_gl_->inputs()) {
     input_shapes_.push_back(input->tensor.shape);

mediapipe/util/tflite/tflite_gpu_runner.h

@@ -54,7 +54,8 @@ class TFLiteGPURunner {
       : options_(options) {}
 
   absl::Status InitializeWithModel(const tflite::FlatBufferModel& flatbuffer,
-                                   const tflite::OpResolver& op_resolver);
+                                   const tflite::OpResolver& op_resolver,
+                                   bool allow_quant_ops = false);
 
   void ForceOpenGL() { opengl_is_forced_ = true; }
   void ForceOpenCL() { opencl_is_forced_ = true; }