commit f23a914b52d1935602d0a94c426ccab15c8fed69
author A. Unique TensorFlower <gardener@tensorflow.org> Fri Nov 22 08:43:35 2019 -0800
committer TensorFlower Gardener <gardener@tensorflow.org> Fri Nov 22 11:44:04 2019 -0800
tree 8b61b7f97cbf17a495dbdfd7920c11b022001b51
parent 70c516fa0d33ee5af9b7ad9bb094a4d0acac6e9e

Use explicit inference priorities instead of setting just allow_precision_loss

PiperOrigin-RevId: 281976010
Change-Id: I5a0056595d81ed4e22f1b7de20cdd93f5d73fd38

tensorflow/lite/tools/benchmark/benchmark_tflite_model.cc

diff --git a/tensorflow/lite/tools/benchmark/benchmark_tflite_model.cc b/tensorflow/lite/tools/benchmark/benchmark_tflite_model.cc
index c3df03c..3448dde 100644
--- a/tensorflow/lite/tools/benchmark/benchmark_tflite_model.cc
+++ b/tensorflow/lite/tools/benchmark/benchmark_tflite_model.cc
@@ -660,8 +660,13 @@
     TfLiteGpuDelegateOptionsV2 gpu_opts = TfLiteGpuDelegateOptionsV2Default();
     gpu_opts.inference_preference =
         TFLITE_GPU_INFERENCE_PREFERENCE_SUSTAINED_SPEED;
-    gpu_opts.is_precision_loss_allowed =
-        params_.Get<bool>("gpu_precision_loss_allowed") ? 1 : 0;
+    if (params_.Get<bool>("gpu_precision_loss_allowed")) {
+      gpu_opts.inference_priority1 = TFLITE_GPU_INFERENCE_PRIORITY_MIN_LATENCY;
+      gpu_opts.inference_priority2 =
+          TFLITE_GPU_INFERENCE_PRIORITY_MIN_MEMORY_USAGE;
+      gpu_opts.inference_priority3 =
+          TFLITE_GPU_INFERENCE_PRIORITY_MAX_PRECISION;
+    }
     Interpreter::TfLiteDelegatePtr delegate =
         evaluation::CreateGPUDelegate(model_.get(), &gpu_opts);
 #elif defined(REAL_IPHONE_DEVICE)