commit 3361681b475eddd88c7a728a674ba5f530621bde
author Viet Dang <vddang@google.com> Tue Feb 12 17:39:05 2019 +0000
committer Viet Dang <vddang@google.com> Wed Feb 13 15:58:21 2019 +0000
tree 7a7a702234c583eeda87b04de76916a0a2e4a161
parent ff86b6083df1a5c6751133d603998574dbfd1249

Adds some getters to BenchmarkResult class.

Bug: 122890757

Test: com.android.nn.benchmark.app.NNTest
Change-Id: I3f7e9601809b9544c06280ba548dbc3cfe111ac1
Merged-In: I3f7e9601809b9544c06280ba548dbc3cfe111ac1
(cherry picked from commit 555f6533f2ccc7a9f44d1df548c19ed6084fced2)

src/com/android/nn/benchmark/core/BenchmarkResult.java

diff --git a/src/com/android/nn/benchmark/core/BenchmarkResult.java b/src/com/android/nn/benchmark/core/BenchmarkResult.java
index 5e4cad6..2a823d4 100644
--- a/src/com/android/nn/benchmark/core/BenchmarkResult.java
+++ b/src/com/android/nn/benchmark/core/BenchmarkResult.java
@@ -20,6 +20,7 @@
 import android.os.Parcel;
 import android.os.Parcelable;
 import android.text.TextUtils;
+import android.util.Pair;
 
 import java.util.ArrayList;
 import java.util.Arrays;
@@ -181,10 +182,22 @@
                 }
             };
 
+    public float getError() {
+        return mSumOfMSEs;
+    }
+
     public float getMeanTimeSec() {
         return mTotalTimeSec / mIterations;
     }
 
+    public List<Pair<String, Float>> getEvaluatorResults() {
+        List<Pair<String, Float>> results = new ArrayList<>();
+        for (int i = 0; i < mEvaluatorKeys.length; ++i) {
+            results.add(new Pair<>(mEvaluatorKeys[i], mEvaluatorResults[i]));
+        }
+        return results;
+    }
+
     @Override
     public String toString() {
         if (!TextUtils.isEmpty(mBenchmarkError)) {
@@ -252,6 +265,7 @@
         return results;
     }
 
+    @SuppressWarnings("AndroidJdkLibsChecker")
     public String toCsvLine() {
         if (!TextUtils.isEmpty(mBenchmarkError)) {
             return "";