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 +