diff options
author | James Ward <james.ward@arm.com> | 2022-08-12 20:48:56 +0100 |
---|---|---|
committer | James Ward <james.ward@arm.com> | 2022-10-11 11:56:02 +0100 |
commit | 8b39043c70332e1e4c95ee6a9616aec40dd3baf1 (patch) | |
tree | fea519246b698eb944b9d58537fc90bc30481d11 /scripts | |
parent | ba5fad356a926d5e1c6e0fe6b546a310230cc5a8 (diff) | |
download | reference_model-8b39043c70332e1e4c95ee6a9616aec40dd3baf1.tar.gz |
Reference model changes for fp16 support
Change-Id: I72f21fcfa153046274969d327313e3349981dbe6
Signed-off-by: James Ward <james.ward@arm.com>
Diffstat (limited to 'scripts')
-rw-r--r-- | scripts/json2numpy/json2numpy.py | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/scripts/json2numpy/json2numpy.py b/scripts/json2numpy/json2numpy.py index 21b1acd..c04013e 100644 --- a/scripts/json2numpy/json2numpy.py +++ b/scripts/json2numpy/json2numpy.py @@ -16,8 +16,10 @@ class NumpyArrayEncoder(json.JSONEncoder): """Encode default operation.""" if isinstance(obj, np.integer): return int(obj) - elif isinstance(obj, np.floating): + elif isinstance(obj, np.float32): return float(obj) + elif isinstance(obj, np.float16): + return np.float16(obj) elif isinstance(obj, np.ndarray): return obj.tolist() return super(NumpyArrayEncoder, self).default(obj) |