diff options
author | Isabella Gottardi <isabella.gottardi@arm.com> | 2019-11-05 17:50:27 +0000 |
---|---|---|
committer | Georgios Pinitas <georgios.pinitas@arm.com> | 2019-11-14 13:32:30 +0000 |
commit | cd4e9abf7a165f15ccd10ac4541365d4f8a6db19 (patch) | |
tree | 442cb14065b866065c1e2837490ae36731a38eef /utils/GraphUtils.cpp | |
parent | ee8cf3287265436d0d12effa36e876dc251eee8c (diff) | |
download | ComputeLibrary-cd4e9abf7a165f15ccd10ac4541365d4f8a6db19.tar.gz |
COMPMID-2452: Add mnist example
* Add small-mnist example
* Add PrintAccessor
* Add DequantizationLayer graph node
Change-Id: I7bc8011e5a602f40fa3c47b231a2a69c804e78c2
Signed-off-by: Isabella Gottardi <isabella.gottardi@arm.com>
Reviewed-on: https://review.mlplatform.org/c/2274
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Giorgio Arena <giorgio.arena@arm.com>
Reviewed-by: Georgios Pinitas <georgios.pinitas@arm.com>
Diffstat (limited to 'utils/GraphUtils.cpp')
-rw-r--r-- | utils/GraphUtils.cpp | 13 |
1 files changed, 13 insertions, 0 deletions
diff --git a/utils/GraphUtils.cpp b/utils/GraphUtils.cpp index e07e26f2fd..71bfc372fe 100644 --- a/utils/GraphUtils.cpp +++ b/utils/GraphUtils.cpp @@ -223,6 +223,19 @@ bool NumPyAccessor::access_tensor(ITensor &tensor) return false; } +#ifdef ARM_COMPUTE_ASSERTS_ENABLED +PrintAccessor::PrintAccessor(std::ostream &output_stream, IOFormatInfo io_fmt) + : _output_stream(output_stream), _io_fmt(io_fmt) +{ +} + +bool PrintAccessor::access_tensor(ITensor &tensor) +{ + tensor.print(_output_stream, _io_fmt); + return false; +} +#endif /* ARM_COMPUTE_ASSERTS_ENABLED */ + SaveNumPyAccessor::SaveNumPyAccessor(std::string npy_name, const bool is_fortran) : _npy_name(std::move(npy_name)), _is_fortran(is_fortran) { |