ml/ml-models-public: Remove intermediate tflite output
as IREE upstream fix has been landed.
BUG: 287490866
Change-Id: Ia9d6c1af5beaab8b98ced9c98fff94950da61911
diff --git a/quant_models/CMakeLists.txt b/quant_models/CMakeLists.txt
index d248bfa..7949575 100644
--- a/quant_models/CMakeLists.txt
+++ b/quant_models/CMakeLists.txt
@@ -28,10 +28,6 @@
"mobilenet_v1_0.25_224_quant.tflite"
C_IDENTIFIER
"quant_models_mobilenet_v1"
- TFLITE_OUTPUT
- # TODO(b/287490866): remove this argument after IREE fix is landed
- "--output-array=MobilenetV1/Logits/SpatialSqueeze"
- MobilenetV1/Logits/SpatialSqueeze
FLAGS
"-iree-input-type=tosa"
"-riscv-v-fixed-length-vector-lmul-max=8"
@@ -56,9 +52,6 @@
"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 557f3bf..d3d40ab 100644
--- a/quant_models/iree_exec/person_detection_test.filecheck
+++ b/quant_models/iree_exec/person_detection_test.filecheck
@@ -1,2 +1,2 @@
-// TODO(b/287490866): change range after IREE fix is landed
-// CHECK: {{Non-person Score: -11[2-4]; Person Score: 11[0-4]}}
+// CHECK: Non-person Score: -113; Person Score: 113
+
