diff options
| author | Slava Shklyaev <slavash@google.com> | 2020-02-12 12:15:01 +0000 |
|---|---|---|
| committer | Slava Shklyaev <slavash@google.com> | 2020-02-18 17:22:21 +0000 |
| commit | 64ffe91670d39eca7e072e07d258a6337c3b3b1b (patch) | |
| tree | 05d9b16a807fb0da4f616b0c520dacba500a9b3f /nn/runtime/test/TestNeuralNetworksWrapper.h | |
| parent | 1f8919ee53d76b2b2f611892a54958cfc282289c (diff) | |
| download | ml-64ffe91670d39eca7e072e07d258a6337c3b3b1b.tar.gz | |
Add setLoopTimeout CTS test
Also tests getDefaultLoopTimeout and getMaximumLoopTimeout.
Bug: 145906499
Bug: 136735929
Test: NNT_static
Change-Id: I1968fe97c6b3f51e9950d17abdc06a764a27c517
Diffstat (limited to 'nn/runtime/test/TestNeuralNetworksWrapper.h')
| -rw-r--r-- | nn/runtime/test/TestNeuralNetworksWrapper.h | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/nn/runtime/test/TestNeuralNetworksWrapper.h b/nn/runtime/test/TestNeuralNetworksWrapper.h index 58dea3e64..ed33161fa 100644 --- a/nn/runtime/test/TestNeuralNetworksWrapper.h +++ b/nn/runtime/test/TestNeuralNetworksWrapper.h @@ -366,6 +366,10 @@ class Execution { mExecution, index, type, memory->get(), offset, length)); } + Result setLoopTimeout(uint64_t duration) { + return static_cast<Result>(ANeuralNetworksExecution_setLoopTimeout(mExecution, duration)); + } + Result startCompute(Event* event) { ANeuralNetworksEvent* ev = nullptr; Result result = static_cast<Result>(ANeuralNetworksExecution_startCompute(mExecution, &ev)); |