IVGCVSW-5200 Update pyarmnn
* Add HalfPixelCenters to Resize
* Update pyarmnn version to semantic versioning
* Add fill operator
* Add Bf16 optimization
* Add Gather operator
* Update TransposeConvolution2d descriptor
* Add Rank operator
* Add load dynamic tensor support of TfLiteParser
Signed-off-by: Jan Eilers <jan.eilers@arm.com>
Change-Id: I7e76ed286ab87bd97a65ff62868ba7db7967376f
diff --git a/python/pyarmnn/test/test_network.py b/python/pyarmnn/test/test_network.py
index fc2591c..679e640 100644
--- a/python/pyarmnn/test/test_network.py
+++ b/python/pyarmnn/test/test_network.py
@@ -6,6 +6,23 @@
import pytest
import pyarmnn as ann
+def test_optimizer_options_default_values():
+ opt = ann.OptimizerOptions()
+ assert opt.m_ReduceFp32ToFp16 == False
+ assert opt.m_Debug == False
+ assert opt.m_ReduceFp32ToBf16 == False
+
+def test_optimizer_options_set_values1():
+ opt = ann.OptimizerOptions(True, True)
+ assert opt.m_ReduceFp32ToFp16 == True
+ assert opt.m_Debug == True
+ assert opt.m_ReduceFp32ToBf16 == False
+
+def test_optimizer_options_set_values2():
+ opt = ann.OptimizerOptions(False, False, True)
+ assert opt.m_ReduceFp32ToFp16 == False
+ assert opt.m_Debug == False
+ assert opt.m_ReduceFp32ToBf16 == True
@pytest.fixture(scope="function")
def get_runtime(shared_data_folder, network_file):
@@ -166,6 +183,7 @@
'AddDivisionLayer',
'AddElementwiseUnaryLayer',
'AddFloorLayer',
+ 'AddFillLayer',
'AddFullyConnectedLayer',
'AddGatherLayer',
'AddInputLayer',
@@ -186,6 +204,7 @@
'AddPreluLayer',
'AddQuantizeLayer',
'AddQuantizedLstmLayer',
+ 'AddRankLayer',
'AddReshapeLayer',
'AddResizeLayer',
'AddSliceLayer',