diff options
author | Sadik Armagan <sadik.armagan@arm.com> | 2020-11-11 18:01:48 +0000 |
---|---|---|
committer | Sadik Armagan <sadik.armagan@arm.com> | 2020-11-12 11:35:11 +0000 |
commit | 4189cc5ca4bb12e02c5e7f86ec6079f76d845b59 (patch) | |
tree | bb5426804692e11abf000ffe9c64f7d95e104beb /delegate/src/test/ArmnnDelegateTest.cpp | |
parent | 8081536d24291794b4e189e6d5532d913a4525cb (diff) | |
download | armnn-4189cc5ca4bb12e02c5e7f86ec6079f76d845b59.tar.gz |
IVGCVSW-5504 'TfLiteDelegate: Introduce FP16 and BackendOptions'
* Added BackendOptions creations of armnn_delegate
* Included armnn/third-party the armnn_delegate unit tests
* Updated the CreateConstTensor function
Signed-off-by: Sadik Armagan <sadik.armagan@arm.com>
Change-Id: I8e2099a465766b905bff701413307e5850b68e42
Diffstat (limited to 'delegate/src/test/ArmnnDelegateTest.cpp')
-rw-r--r-- | delegate/src/test/ArmnnDelegateTest.cpp | 13 |
1 files changed, 11 insertions, 2 deletions
diff --git a/delegate/src/test/ArmnnDelegateTest.cpp b/delegate/src/test/ArmnnDelegateTest.cpp index 7cec70b022..050fe45ca0 100644 --- a/delegate/src/test/ArmnnDelegateTest.cpp +++ b/delegate/src/test/ArmnnDelegateTest.cpp @@ -35,9 +35,18 @@ TEST_CASE ("ArmnnDelegate Registered") const TfLiteRegistration* opRegister = opResolver.FindOp(BuiltinOperator_ADD, 1); tfLiteInterpreter->AddNodeWithParameters({0, 1}, {2}, "", 0, nullptr, opRegister); - // create the Armnn Delegate + // Create the Armnn Delegate std::vector<armnn::BackendId> backends = { armnn::Compute::CpuRef }; - armnnDelegate::DelegateOptions delegateOptions(backends); + std::vector<armnn::BackendOptions> backendOptions; + backendOptions.emplace_back( + armnn::BackendOptions{ "BackendName", + { + { "Option1", 42 }, + { "Option2", true } + }} + ); + + armnnDelegate::DelegateOptions delegateOptions(backends, backendOptions); std::unique_ptr<TfLiteDelegate, decltype(&armnnDelegate::TfLiteArmnnDelegateDelete)> theArmnnDelegate(armnnDelegate::TfLiteArmnnDelegateCreate(delegateOptions), armnnDelegate::TfLiteArmnnDelegateDelete); |