diff options
author | Nattapat Chaimanowong <nattapat.chaimanowong@arm.com> | 2019-04-04 13:33:10 +0100 |
---|---|---|
committer | Nattapat Chaimanowong <nattapat.chaimanowong@arm.com> | 2019-04-04 13:33:10 +0100 |
commit | d5fd9767d426ca465eb660be062eecce51ad1097 (patch) | |
tree | b2a2b684c9072dda7b52b40e579a5b0c4aae3c21 /ArmnnDriverImpl.cpp | |
parent | 16196e267833178dae62926ff090c50ec6813ad4 (diff) | |
download | android-nn-driver-d5fd9767d426ca465eb660be062eecce51ad1097.tar.gz |
IVGCVSW-2886 Support multiple backends in Android driver
Change-Id: I4abe1f750801911570b6dc65c187b828c5929b5f
Signed-off-by: Nattapat Chaimanowong <nattapat.chaimanowong@arm.com>
Diffstat (limited to 'ArmnnDriverImpl.cpp')
-rw-r--r-- | ArmnnDriverImpl.cpp | 14 |
1 files changed, 7 insertions, 7 deletions
diff --git a/ArmnnDriverImpl.cpp b/ArmnnDriverImpl.cpp index f6456ee1..40bd80ab 100644 --- a/ArmnnDriverImpl.cpp +++ b/ArmnnDriverImpl.cpp @@ -77,9 +77,9 @@ Return<void> ArmnnDriverImpl<HalPolicy>::getSupportedOperations(const armnn::IRu } // Attempt to convert the model to an ArmNN input network (INetwork). - ModelToINetworkConverter<HalPolicy> modelConverter(options.GetComputeDevice(), - model, - options.GetForcedUnsupportedOperations()); + ModelToINetworkConverter<HalPolicy> modelConverter(options.GetBackends(), + model, + options.GetForcedUnsupportedOperations()); if (modelConverter.GetConversionResult() != ConversionResult::Success && modelConverter.GetConversionResult() != ConversionResult::UnsupportedFeature) @@ -132,9 +132,9 @@ Return<ErrorStatus> ArmnnDriverImpl<HalPolicy>::prepareModel( // at this point we're being asked to prepare a model that we've already declared support for // and the operation indices may be different to those in getSupportedOperations anyway. set<unsigned int> unsupportedOperations; - ModelToINetworkConverter<HalPolicy> modelConverter(options.GetComputeDevice(), - model, - unsupportedOperations); + ModelToINetworkConverter<HalPolicy> modelConverter(options.GetBackends(), + model, + unsupportedOperations); if (modelConverter.GetConversionResult() != ConversionResult::Success) { @@ -151,7 +151,7 @@ Return<ErrorStatus> ArmnnDriverImpl<HalPolicy>::prepareModel( try { optNet = armnn::Optimize(*modelConverter.GetINetwork(), - {options.GetComputeDevice()}, + options.GetBackends(), runtime->GetDeviceSpec(), OptOptions, errMessages); |