diff options
author | Slava Shklyaev <slavash@google.com> | 2020-03-04 17:15:05 +0000 |
---|---|---|
committer | Slava Shklyaev <slavash@google.com> | 2020-03-09 12:07:16 +0000 |
commit | 5573eefc99585c9438629c9e006942d1345f3651 (patch) | |
tree | f2a75e523893c750e5de3c116b911cc3c10e9cda /nn/runtime/test/TestNeuralNetworksWrapper.h | |
parent | c0c67a23e356d4a0e1b2b9aa2264441bfa7565c4 (diff) | |
download | ml-5573eefc99585c9438629c9e006942d1345f3651.tar.gz |
Add test_wrapper::Compilation::createForDevice()
Bug: 147105700
Test: m
Change-Id: I8d2b148c91b0b79d067b7f3020b42958163f7856
Merged-In: I8d2b148c91b0b79d067b7f3020b42958163f7856
(cherry picked from commit 8ee0edf5f65ecb72189d215f0f32c6349acf1897)
Diffstat (limited to 'nn/runtime/test/TestNeuralNetworksWrapper.h')
-rw-r--r-- | nn/runtime/test/TestNeuralNetworksWrapper.h | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/nn/runtime/test/TestNeuralNetworksWrapper.h b/nn/runtime/test/TestNeuralNetworksWrapper.h index ed33161fa..53ba5d291 100644 --- a/nn/runtime/test/TestNeuralNetworksWrapper.h +++ b/nn/runtime/test/TestNeuralNetworksWrapper.h @@ -271,6 +271,11 @@ class Compilation { return *this; } + Result createForDevice(const Model* model, const ANeuralNetworksDevice* device) { + return static_cast<Result>(ANeuralNetworksCompilation_createForDevices( + model->getHandle(), &device, 1, &mCompilation)); + } + Result setPreference(ExecutePreference preference) { return static_cast<Result>(ANeuralNetworksCompilation_setPreference( mCompilation, static_cast<int32_t>(preference))); |