diff options
author | Teresa Charlin <teresa.charlinreyes@arm.com> | 2023-03-23 15:28:02 +0000 |
---|---|---|
committer | TeresaARM <teresa.charlinreyes@arm.com> | 2023-04-19 08:54:57 +0000 |
commit | 93f0ad0016a6450670fbf650568f5724c7bbb63e (patch) | |
tree | efb5f687ff93bf34bae1fe9d218124b57c2681ac /delegate/test/DelegateOptionsTestHelper.hpp | |
parent | acb3ec51e51542d3011ed87842f87c2261abaaff (diff) | |
download | armnn-93f0ad0016a6450670fbf650568f5724c7bbb63e.tar.gz |
GitHub #640 Add support for CEIL operator
* Reference workload
* TfLite Delegate
* TfLite Parser
* Serializer and Deserializer
* Changed fallback tests in delegate to use COS instead of CEIL
Signed-off-by: Teresa Charlin <teresa.charlinreyes@arm.com>
Signed-off-by: Mike Kelly <mike.kelly@arm.com>
Change-Id: I36e0dbff33694182d1dba0c95d463506428e2f04
Diffstat (limited to 'delegate/test/DelegateOptionsTestHelper.hpp')
-rw-r--r-- | delegate/test/DelegateOptionsTestHelper.hpp | 18 |
1 files changed, 9 insertions, 9 deletions
diff --git a/delegate/test/DelegateOptionsTestHelper.hpp b/delegate/test/DelegateOptionsTestHelper.hpp index b6974c9fb6..148a6d2c22 100644 --- a/delegate/test/DelegateOptionsTestHelper.hpp +++ b/delegate/test/DelegateOptionsTestHelper.hpp @@ -152,10 +152,10 @@ std::vector<char> CreateAddDivTfLiteModel(tflite::TensorType tensorType, flatBufferBuilder.GetBufferPointer() + flatBufferBuilder.GetSize()); } -std::vector<char> CreateCeilTfLiteModel(tflite::TensorType tensorType, - const std::vector <int32_t>& tensorShape, - float quantScale = 1.0f, - int quantOffset = 0) +std::vector<char> CreateCosTfLiteModel(tflite::TensorType tensorType, + const std::vector <int32_t>& tensorShape, + float quantScale = 1.0f, + int quantOffset = 0) { using namespace tflite; flatbuffers::FlatBufferBuilder flatBufferBuilder; @@ -199,7 +199,7 @@ std::vector<char> CreateCeilTfLiteModel(tflite::TensorType tensorType, flatbuffers::Offset<flatbuffers::String> modelDescription = flatBufferBuilder.CreateString("ArmnnDelegate: CEIL Operator Model"); flatbuffers::Offset<OperatorCode> operatorCode = - CreateOperatorCode(flatBufferBuilder, tflite::BuiltinOperator_CEIL); + CreateOperatorCode(flatBufferBuilder, tflite::BuiltinOperator_COS); const std::vector<int32_t> subgraphInputs({0}); const std::vector<int32_t> subgraphOutputs({1}); @@ -277,10 +277,10 @@ void DelegateOptionNoFallbackTest(tflite::TensorType tensorType, int quantOffset = 0) { using namespace delegateTestInterpreter; - std::vector<char> modelBuffer = CreateCeilTfLiteModel(tensorType, - tensorShape, - quantScale, - quantOffset); + std::vector<char> modelBuffer = CreateCosTfLiteModel(tensorType, + tensorShape, + quantScale, + quantOffset); // Setup interpreter with just TFLite Runtime. auto tfLiteInterpreter = DelegateTestInterpreter(modelBuffer); |