diff options
author | 2018-02-14 17:51:17 -0800 | |
---|---|---|
committer | 2018-02-14 17:55:32 -0800 | |
commit | 61207fad1557da22a0d06bb22e35f23bd9ca4938 (patch) | |
tree | b1ed41f647d75ee11d92cea693da1bc0c9a658ff /tensorflow/contrib/lite/testing/test_runner.h | |
parent | 5b3b689cc307632d38f08b1c7f6952bbe7a7ad7e (diff) |
Extract ReadOutput method to test runner
PiperOrigin-RevId: 185773920
Diffstat (limited to 'tensorflow/contrib/lite/testing/test_runner.h')
-rw-r--r-- | tensorflow/contrib/lite/testing/test_runner.h | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/tensorflow/contrib/lite/testing/test_runner.h b/tensorflow/contrib/lite/testing/test_runner.h index 60eaafa474..05770beee2 100644 --- a/tensorflow/contrib/lite/testing/test_runner.h +++ b/tensorflow/contrib/lite/testing/test_runner.h @@ -68,6 +68,10 @@ class TestRunner { // satisfied. virtual bool CheckResults() = 0; + // Read contents of tensor into csv format. + // The given 'id' is guaranteed to be one of the ids returned by GetOutputs(). + virtual string ReadOutput(int id) = 0; + // Set the base path for loading models. void SetModelBaseDir(const string& path) { model_base_dir_ = path; |