ml/ml-models-public: use intermediate output for person_detection model
Change test output range accordingly.
Change-Id: Ieab7d8f8683721ec81903973ad9590d6d56b7289
diff --git a/quant_models/CMakeLists.txt b/quant_models/CMakeLists.txt
index e8dec30..b6fef98 100644
--- a/quant_models/CMakeLists.txt
+++ b/quant_models/CMakeLists.txt
@@ -52,6 +52,9 @@
"person_detection.tflite"
C_IDENTIFIER
"quant_models_person_detection"
+ TFLITE_OUTPUT
+ # TODO(b/287490866): remove this argument after IREE fix is landed
+ "--output-array=MobilenetV1/Logits/SpatialSqueeze"
FLAGS
"-iree-input-type=tosa"
"-riscv-v-fixed-length-vector-lmul-max=8"
diff --git a/quant_models/iree_exec/person_detection_test.filecheck b/quant_models/iree_exec/person_detection_test.filecheck
index 1a880de..557f3bf 100644
--- a/quant_models/iree_exec/person_detection_test.filecheck
+++ b/quant_models/iree_exec/person_detection_test.filecheck
@@ -1 +1,2 @@
-// CHECK: {{Non-person Score: -11[3-4]; Person Score: 11[3-4]}}
+// TODO(b/287490866): change range after IREE fix is landed
+// CHECK: {{Non-person Score: -11[2-4]; Person Score: 11[0-4]}}
