aboutsummaryrefslogtreecommitdiff
path: root/delegate/src/test/ElementwiseBinaryTestHelper.hpp
diff options
context:
space:
mode:
authorJim Flynn <jim.flynn@arm.com>2021-10-13 21:20:07 +0100
committerMatthew Sloyan <matthew.sloyan@arm.com>2021-11-01 12:09:27 +0000
commit4b2f34709be018d6cf9931b66deaf84a4469340d (patch)
tree2320ebf3ee3aeb91b91fb55de16504bd80f6f3a3 /delegate/src/test/ElementwiseBinaryTestHelper.hpp
parent9f6862de94e3d15ea5207a5747012f6c7eead358 (diff)
downloadarmnn-4b2f34709be018d6cf9931b66deaf84a4469340d.tar.gz
IVGCVSW-6457 Add FLOOR_DIV Support to the TfLiteDelegate
Change-Id: Ia4bf42b1f3f86b947825dff8e538d2d4343effab Signed-off-by: Jim Flynn <jim.flynn@arm.com>
Diffstat (limited to 'delegate/src/test/ElementwiseBinaryTestHelper.hpp')
-rw-r--r--delegate/src/test/ElementwiseBinaryTestHelper.hpp6
1 files changed, 6 insertions, 0 deletions
diff --git a/delegate/src/test/ElementwiseBinaryTestHelper.hpp b/delegate/src/test/ElementwiseBinaryTestHelper.hpp
index 13b336e91e..69b0c88dc8 100644
--- a/delegate/src/test/ElementwiseBinaryTestHelper.hpp
+++ b/delegate/src/test/ElementwiseBinaryTestHelper.hpp
@@ -123,6 +123,12 @@ std::vector<char> CreateElementwiseBinaryTfLiteModel(tflite::BuiltinOperator bin
operatorBuiltinOptions = CreateSubOptions(flatBufferBuilder, activationType).Union();
break;
}
+ case BuiltinOperator_FLOOR_DIV:
+ {
+ operatorBuiltinOptionsType = tflite::BuiltinOptions_FloorDivOptions;
+ operatorBuiltinOptions = CreateSubOptions(flatBufferBuilder, activationType).Union();
+ break;
+ }
default:
break;
}