diff options
| author | Isabella Gottardi <isabella.gottardi@arm.com> | 2019-07-16 11:48:51 +0100 |
|---|---|---|
| committer | Manuel Bottini <manuel.bottini@arm.com> | 2019-07-16 16:34:59 +0000 |
| commit | 2ea3761416aab259d9d84620dba2e011bcb5d880 (patch) | |
| tree | 61dc530fb18cf83ee45286837326d6be0196fcfa /utils/GraphUtils.cpp | |
| parent | ba27e4467dfc04e23ce9483330be062e9aaebdc5 (diff) | |
| download | ComputeLibrary-2ea3761416aab259d9d84620dba2e011bcb5d880.tar.gz | |
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>
Diffstat (limited to 'utils/GraphUtils.cpp')
| -rw-r--r-- | utils/GraphUtils.cpp | 14 |
1 files changed, 14 insertions, 0 deletions
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)) { |