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/CL/kernels/CLDirectConvolutionLayerKernel.cpp | |
| 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/CL/kernels/CLDirectConvolutionLayerKernel.cpp')
| -rw-r--r-- | src/core/CL/kernels/CLDirectConvolutionLayerKernel.cpp | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/src/core/CL/kernels/CLDirectConvolutionLayerKernel.cpp b/src/core/CL/kernels/CLDirectConvolutionLayerKernel.cpp index 9d1c87d9e1..265c5074c5 100644 --- a/src/core/CL/kernels/CLDirectConvolutionLayerKernel.cpp +++ b/src/core/CL/kernels/CLDirectConvolutionLayerKernel.cpp @@ -246,7 +246,7 @@ void CLDirectConvolutionLayerKernel::run(const Window &window, cl::CommandQueue if(_biases != nullptr) { Window slice_biases; - slice_biases.use_tensor_dimensions(_biases->info()); + slice_biases.use_tensor_dimensions(_biases->info()->tensor_shape()); add_1D_tensor_argument(idx1, _biases, slice_biases); } |