commit dbd14eb7bcfc3255455c343da1b52f99bbd864ff
author Cody Heiner <codyheiner@google.com> Thu Mar 30 18:41:45 2023 -0700
committer Cody Heiner <codyheiner@google.com> Wed Apr 05 19:31:11 2023 -0700
tree dcd7cdebf8a35e34f905ab111475e1f39f8cb68c
parent 1f428c8affd9ca3951b01e88cb93545f109f78f3

Add outputLength method

Test: build succeeds, `atest libinput-tests` passes.

Bug: 268245099

Change-Id: I030da703a907eef44e85d186144eddc53b5998cc

libs/input/TfLiteMotionPredictor.cpp

diff --git a/libs/input/TfLiteMotionPredictor.cpp b/libs/input/TfLiteMotionPredictor.cpp
index 3b061d1..85fa176 100644
--- a/libs/input/TfLiteMotionPredictor.cpp
+++ b/libs/input/TfLiteMotionPredictor.cpp
@@ -346,6 +346,10 @@
     return getTensorBuffer<const float>(mInputR).size();
 }
 
+size_t TfLiteMotionPredictorModel::outputLength() const {
+    return getTensorBuffer<const float>(mOutputR).size();
+}
+
 std::span<float> TfLiteMotionPredictorModel::inputR() {
     return getTensorBuffer<float>(mInputR);
 }

libs/input/tests/TfLiteMotionPredictor_test.cpp

diff --git a/libs/input/tests/TfLiteMotionPredictor_test.cpp b/libs/input/tests/TfLiteMotionPredictor_test.cpp
index 6e76ac1..b5ed9e4 100644
--- a/libs/input/tests/TfLiteMotionPredictor_test.cpp
+++ b/libs/input/tests/TfLiteMotionPredictor_test.cpp
@@ -139,8 +139,10 @@
 
     ASSERT_TRUE(model->invoke());
 
-    ASSERT_EQ(model->outputR().size(), model->outputPhi().size());
-    ASSERT_EQ(model->outputR().size(), model->outputPressure().size());
+    const int outputLength = model->outputLength();
+    ASSERT_EQ(outputLength, model->outputR().size());
+    ASSERT_EQ(outputLength, model->outputPhi().size());
+    ASSERT_EQ(outputLength, model->outputPressure().size());
 }
 
 TEST(TfLiteMotionPredictorTest, ModelOutput) {