diff options
author | SiCong Li <sicong.li@arm.com> | 2017-08-23 11:02:43 +0100 |
---|---|---|
committer | Anthony Barbier <anthony.barbier@arm.com> | 2018-11-02 16:35:24 +0000 |
commit | 86b53339679e12c952a24a8845a5409ac3d52de6 (patch) | |
tree | 807c897ca1001f22b1906d285488877a287b482b /src/core/NEON | |
parent | 70e9bc21682f4eaedaceb632f594f588cb2c91fc (diff) | |
download | ComputeLibrary-86b53339679e12c952a24a8845a5409ac3d52de6.tar.gz |
COMPMID-514 (3RDPARTY_UPDATE)(DATA_UPDATE) Add support to load .npy data
* Add tensorflow_data_extractor script.
* Incorporate 3rdparty npy reader libnpy.
* Port AlexNet system test to validation_new.
* Port LeNet5 system test to validation_new.
* Update 3rdparty/ and data/ submodules.
Change-Id: I156d060fe9185cd8db810b34bf524cbf5cb34f61
Reviewed-on: http://mpd-gerrit.cambridge.arm.com/84914
Reviewed-by: Anthony Barbier <anthony.barbier@arm.com>
Tested-by: Kaizen <jeremy.johnson+kaizengerrit@arm.com>
Diffstat (limited to 'src/core/NEON')
-rw-r--r-- | src/core/NEON/kernels/NEFillBorderKernel.cpp | 2 | ||||
-rw-r--r-- | src/core/NEON/kernels/NEFillInnerBorderKernel.cpp | 2 | ||||
-rw-r--r-- | src/core/NEON/kernels/NEIm2ColKernel.cpp | 2 |
3 files changed, 3 insertions, 3 deletions
diff --git a/src/core/NEON/kernels/NEFillBorderKernel.cpp b/src/core/NEON/kernels/NEFillBorderKernel.cpp index 65d5388c4b..3f1f678a7e 100644 --- a/src/core/NEON/kernels/NEFillBorderKernel.cpp +++ b/src/core/NEON/kernels/NEFillBorderKernel.cpp @@ -120,7 +120,7 @@ void NEFillBorderKernel::configure(ITensor *tensor, BorderSize border_size, Bord Window win; win.set(Window::DimX, Window::Dimension(0, 1, 1)); win.set(Window::DimY, Window::Dimension(0, 1, 1)); - win.use_tensor_dimensions(_tensor->info(), Window::DimZ); + win.use_tensor_dimensions(_tensor->info()->tensor_shape(), Window::DimZ); INEKernel::configure(win); } diff --git a/src/core/NEON/kernels/NEFillInnerBorderKernel.cpp b/src/core/NEON/kernels/NEFillInnerBorderKernel.cpp index 5323733fd3..017e259ca4 100644 --- a/src/core/NEON/kernels/NEFillInnerBorderKernel.cpp +++ b/src/core/NEON/kernels/NEFillInnerBorderKernel.cpp @@ -57,7 +57,7 @@ void NEFillInnerBorderKernel::configure(ITensor *input, BorderSize border_size, Window win; win.set(Window::DimX, Window::Dimension(0, 1, 1)); win.set(Window::DimY, Window::Dimension(0, 1, 1)); - win.use_tensor_dimensions(_tensor->info(), Window::DimZ); + win.use_tensor_dimensions(_tensor->info()->tensor_shape(), Window::DimZ); INEKernel::configure(win); } diff --git a/src/core/NEON/kernels/NEIm2ColKernel.cpp b/src/core/NEON/kernels/NEIm2ColKernel.cpp index 3e50277cdf..71910e3a69 100644 --- a/src/core/NEON/kernels/NEIm2ColKernel.cpp +++ b/src/core/NEON/kernels/NEIm2ColKernel.cpp @@ -231,7 +231,7 @@ void NEIm2ColKernel::run_reduced(const Window &window) in_window.set(Window::DimX, Window::Dimension(0, 1, 1)); Window out_window; - out_window.use_tensor_dimensions(_output->info()); + out_window.use_tensor_dimensions(_output->info()->tensor_shape()); out_window.set(Window::DimX, Window::Dimension(out_window.x().start(), out_window.x().end(), in_width)); Window in_slice = in_window.first_slice_window_3D(); |