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);
 }