diff options
author | Narumol Prangnawarat <narumol.prangnawarat@arm.com> | 2021-01-29 15:38:54 +0000 |
---|---|---|
committer | Narumol Prangnawarat <narumol.prangnawarat@arm.com> | 2021-02-02 13:53:38 +0000 |
commit | 74a3cf5755b801cf258177e8e55b4cda64a0c351 (patch) | |
tree | 01246003e2b336e75095d148000370a17e790511 /delegate/src/test/DelegateOptionsTest.cpp | |
parent | bd4fcc30a3732ce63e100d556ee5c9ea9e556c05 (diff) | |
download | armnn-74a3cf5755b801cf258177e8e55b4cda64a0c351.tar.gz |
IVGCVSW-5619 Enable OptimizerOptions for the python external delegate
* Add reduce-fp32-to-fp16, reduce-fp32-to-bf16, debug-data, memory-import
options to external delegate
* Simplify DelegateOptions
* Add test mock models
* Unit tests
* Configure lfs to manage tflite files
Signed-off-by: Narumol Prangnawarat <narumol.prangnawarat@arm.com>
Change-Id: I1e4db468862ba03d4cb031347bc307cf940b3cb1
Diffstat (limited to 'delegate/src/test/DelegateOptionsTest.cpp')
-rw-r--r-- | delegate/src/test/DelegateOptionsTest.cpp | 12 |
1 files changed, 4 insertions, 8 deletions
diff --git a/delegate/src/test/DelegateOptionsTest.cpp b/delegate/src/test/DelegateOptionsTest.cpp index c623781301..23510c7777 100644 --- a/delegate/src/test/DelegateOptionsTest.cpp +++ b/delegate/src/test/DelegateOptionsTest.cpp @@ -25,8 +25,7 @@ TEST_CASE ("ArmnnDelegateOptimizerOptionsReduceFp32ToFp16") // Enable ReduceFp32ToFp16 armnn::OptimizerOptions optimizerOptions(true, true, false, false); - armnn::INetworkProperties networkProperties; - armnnDelegate::DelegateOptions delegateOptions(backends, optimizerOptions, networkProperties); + armnnDelegate::DelegateOptions delegateOptions(backends, optimizerOptions); DelegateOptionTest<float>(::tflite::TensorType_FLOAT32, backends, @@ -56,8 +55,7 @@ TEST_CASE ("ArmnnDelegateOptimizerOptionsDebug") // Enable Debug armnn::OptimizerOptions optimizerOptions(false, true, false, false); - armnn::INetworkProperties networkProperties; - armnnDelegate::DelegateOptions delegateOptions(backends, optimizerOptions, networkProperties); + armnnDelegate::DelegateOptions delegateOptions(backends, optimizerOptions); DelegateOptionTest<float>(::tflite::TensorType_FLOAT32, backends, @@ -98,7 +96,6 @@ TEST_CASE ("ArmnnDelegateOptimizerOptionsDebugFunction") armnn::INetworkProperties networkProperties; armnnDelegate::DelegateOptions delegateOptions(backends, optimizerOptions, - networkProperties, armnn::EmptyOptional(), armnn::Optional<armnn::DebugCallbackFunction>(mockCallback)); @@ -136,11 +133,10 @@ TEST_CASE ("ArmnnDelegateOptimizerOptionsImport") std::vector<int32_t> tensorShape { 1, 2, 2, 1 }; std::vector<uint8_t> inputData = { 1, 2, 3, 4 }; std::vector<uint8_t> divData = { 2, 2, 3, 4 }; - std::vector<uint8_t> expectedResult = { 1, 2, 2, 2}; + std::vector<uint8_t> expectedResult = { 1, 2, 2, 2 }; armnn::OptimizerOptions optimizerOptions(false, false, false, true); - armnn::INetworkProperties networkProperties(true, true); - armnnDelegate::DelegateOptions delegateOptions(backends, optimizerOptions, networkProperties); + armnnDelegate::DelegateOptions delegateOptions(backends, optimizerOptions); DelegateOptionTest<uint8_t>(::tflite::TensorType_UINT8, backends, |