aboutsummaryrefslogtreecommitdiff
path: root/python/pyarmnn/test/test_network.py
diff options
context:
space:
mode:
authorRyan OShea <ryan.oshea3@arm.com>2021-11-18 16:52:41 +0000
committerTeresaARM <teresa.charlinreyes@arm.com>2021-11-26 10:27:39 +0000
commit09a052251b3c4c14c093e79d83c36504746890be (patch)
treedb70f9e26250cf220db44265ec159878e53a1aa6 /python/pyarmnn/test/test_network.py
parente0d1fc5ad4f47bfdff2fe2f3fbcebf68669257dd (diff)
downloadarmnn-09a052251b3c4c14c093e79d83c36504746890be.tar.gz
IVGCVSW-6589 AddReduceLayer to PyArmNN
* AddReduceLayer to PyArmNN armnn_network.i * AddReduceDescriptor to PyArmNN armnn_descriptors.i * Add layer to test_network.py * Add descriptor unit tests to test_descriptors.py Signed-off-by: Ryan OShea <ryan.oshea3@arm.com> Change-Id: I2a8706052baf8f9c0038a5ea327d73b8f3ea1961
Diffstat (limited to 'python/pyarmnn/test/test_network.py')
-rw-r--r--python/pyarmnn/test/test_network.py1
1 files changed, 1 insertions, 0 deletions
diff --git a/python/pyarmnn/test/test_network.py b/python/pyarmnn/test/test_network.py
index e33470e090..e56621a23b 100644
--- a/python/pyarmnn/test/test_network.py
+++ b/python/pyarmnn/test/test_network.py
@@ -229,6 +229,7 @@ def test_serialize_to_dot_mode_readonly(network_file, get_runtime, tmpdir):
'AddQuantizeLayer',
'AddQuantizedLstmLayer',
'AddRankLayer',
+ 'AddReduceLayer',
'AddReshapeLayer',
'AddResizeLayer',
'AddShapeLayer',