From 2ea3761416aab259d9d84620dba2e011bcb5d880 Mon Sep 17 00:00:00 2001
From: Isabella Gottardi <isabella.gottardi@arm.com>
Date: Tue, 16 Jul 2019 11:48:51 +0100
Subject: COMPMID-2336: Fix InPlaceMutator condition and add SaveNumpyAccessor

Change-Id: I223a688cfc19465f8581f691b32891cefd375907
Signed-off-by: Isabella Gottardi <isabella.gottardi@arm.com>
Reviewed-on: https://review.mlplatform.org/c/1555
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Pablo Marquez <pablo.tello@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Michalis Spyrou <michalis.spyrou@arm.com>
---
 utils/GraphUtils.cpp | 14 ++++++++++++++
 1 file changed, 14 insertions(+)

(limited to 'utils/GraphUtils.cpp')

diff --git a/utils/GraphUtils.cpp b/utils/GraphUtils.cpp
index 6be289a7e1..dad9aed6a5 100644
--- a/utils/GraphUtils.cpp
+++ b/utils/GraphUtils.cpp
@@ -184,6 +184,20 @@ bool NumPyAccessor::access_tensor(ITensor &tensor)
     return false;
 }
 
+SaveNumPyAccessor::SaveNumPyAccessor(std::string npy_name, const bool is_fortran)
+    : _npy_name(std::move(npy_name)), _is_fortran(is_fortran)
+{
+}
+
+bool SaveNumPyAccessor::access_tensor(ITensor &tensor)
+{
+    ARM_COMPUTE_ERROR_ON_DATA_TYPE_CHANNEL_NOT_IN(&tensor, 1, DataType::F32);
+
+    utils::save_to_npy(tensor, _npy_name, _is_fortran);
+
+    return false;
+}
+
 ImageAccessor::ImageAccessor(std::string filename, bool bgr, std::unique_ptr<IPreprocessor> preprocessor)
     : _already_loaded(false), _filename(std::move(filename)), _bgr(bgr), _preprocessor(std::move(preprocessor))
 {
-- 
cgit v1.2.1