From 2e4d889fb036d1c0a34503400a3f45cfc6f9f3e1 Mon Sep 17 00:00:00 2001
From: Matthew Sloyan <matthew.sloyan@arm.com>
Date: Tue, 18 Oct 2022 18:02:48 +0100
Subject: Add FP16 support to IModelRunner

 * Added specific FP16 readfromVector and writeToVector methods.
 * Added FP16 support to float readfromVector and writeToVector methods.
 * Added missing reference to IModelRunner::setInput.

Signed-off-by: Matthew Sloyan <matthew.sloyan@arm.com>
Change-Id: I6b66468737e672afc925ccad4fb710fbb9427c14
---
 reference_model/src/tensor.h | 2 ++
 1 file changed, 2 insertions(+)

(limited to 'reference_model/src/tensor.h')

diff --git a/reference_model/src/tensor.h b/reference_model/src/tensor.h
index efd7e62..4efbf84 100644
--- a/reference_model/src/tensor.h
+++ b/reference_model/src/tensor.h
@@ -229,11 +229,13 @@ public:
     virtual int copyValueFrom(Tensor* tensor) = 0;
 
     virtual int readfromVector(const std::vector<float>& vals);
+    virtual int readfromVector(const std::vector<half_float::half>& vals);
     virtual int readfromVector(const std::vector<int32_t>& vals);
     virtual int readfromVector(const std::vector<int64_t>& vals);
     virtual int readfromVector(const std::vector<unsigned char>& vals);
 
     virtual int writeToVector(std::vector<float>& vals);
+    virtual int writeToVector(std::vector<half_float::half>& vals);
     virtual int writeToVector(std::vector<int32_t>& vals);
     virtual int writeToVector(std::vector<int64_t>& vals);
     virtual int writeToVector(std::vector<unsigned char>& vals);
-- 
cgit v1.2.1