aboutsummaryrefslogtreecommitdiff
path: root/python/pyarmnn/test
diff options
context:
space:
mode:
Diffstat (limited to 'python/pyarmnn/test')
-rw-r--r--python/pyarmnn/test/test_network.py3
1 files changed, 3 insertions, 0 deletions
diff --git a/python/pyarmnn/test/test_network.py b/python/pyarmnn/test/test_network.py
index 91602b8fa8..933828329b 100644
--- a/python/pyarmnn/test/test_network.py
+++ b/python/pyarmnn/test/test_network.py
@@ -193,6 +193,7 @@ def test_serialize_to_dot_mode_readonly(network_file, get_runtime, tmpdir):
'AddBatchMatMulLayer',
'AddBatchNormalizationLayer',
'AddBatchToSpaceNdLayer',
+ 'AddBroadcastToLayer',
'AddCastLayer',
'AddChannelShuffleLayer',
'AddComparisonLayer',
@@ -210,6 +211,7 @@ def test_serialize_to_dot_mode_readonly(network_file, get_runtime, tmpdir):
'AddFloorLayer',
'AddFillLayer',
'AddFullyConnectedLayer',
+ 'AddFusedLayer',
'AddGatherLayer',
'AddGatherNdLayer',
'AddInputLayer',
@@ -237,6 +239,7 @@ def test_serialize_to_dot_mode_readonly(network_file, get_runtime, tmpdir):
'AddReshapeLayer',
'AddResizeLayer',
'AddReverseV2Layer',
+ 'AddScatterNdLayer',
'AddShapeLayer',
'AddSliceLayer',
'AddSoftmaxLayer',