1 files changed, 11 insertions, 1 deletions
diff --git a/reference_model/src/model_runner.cc b/reference_model/src/model_runner.cc
index 5c086e6..62d6ce6 100644
--- a/reference_model/src/model_runner.cc
+++ b/reference_model/src/model_runner.cc
@@ -55,7 +55,12 @@ GraphStatus IModelRunner::run()
 template <typename T>
 int IModelRunner::setInput(std::string input_name, std::vector<T>& vals)
 {
-    return model_runner_impl->setInput<T>(input_name, vals);
+    return model_runner_impl->setInput<T>(input_name, ArrayProxy(vals.size(), vals.data()));
+}
+
+int IModelRunner::setInput(std::string input_name, uint8_t* raw_ptr, size_t size)
+{
+    return model_runner_impl->setInput(input_name, raw_ptr, size);
 }
 
 template <typename T>
@@ -64,6 +69,11 @@ std::vector<T> IModelRunner::getOutput(std::string output_name)
     return model_runner_impl->getOutput<T>(output_name);
 }
 
+int IModelRunner::getOutput(std::string output_name, uint8_t* raw_ptr, size_t size)
+{
+    return model_runner_impl->getOutput(output_name, raw_ptr, size);
+}
+
 // Template explicit specialization
 template int IModelRunner::setInput<float>(std::string input_name, std::vector<float>& vals);
 template int IModelRunner::setInput<half_float::half>(std::string input_name, std::vector<half_float::half>& vals);