diff options
author | Andrew Mundy <andrew.mundy@arm.com> | 2018-03-15 16:47:03 +0000 |
---|---|---|
committer | Anthony Barbier <anthony.barbier@arm.com> | 2018-11-02 16:49:16 +0000 |
commit | 4d9379a9d3ada794f532ce8acdc8607f4faa2b21 (patch) | |
tree | 14ba02ebcdaf6cb927e9422e45cbab6456c9a097 /src/core/NEON/kernels/NEWinogradLayerKernel.cpp | |
parent | 3f217ec4ff11e20fe686beb9a28d0bbd80a56cd6 (diff) | |
download | ComputeLibrary-4d9379a9d3ada794f532ce8acdc8607f4faa2b21.tar.gz |
COMPMID-1040: Added support for nullptr bias tensor in NEWinogradLayer
Change-Id: Ie624ee17c63dede711d913a82819e128954a57c9
Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/124861
Tested-by: Jenkins <bsgcomp@arm.com>
Reviewed-by: Anthony Barbier <anthony.barbier@arm.com>
Diffstat (limited to 'src/core/NEON/kernels/NEWinogradLayerKernel.cpp')
-rw-r--r-- | src/core/NEON/kernels/NEWinogradLayerKernel.cpp | 3 |
1 files changed, 1 insertions, 2 deletions
diff --git a/src/core/NEON/kernels/NEWinogradLayerKernel.cpp b/src/core/NEON/kernels/NEWinogradLayerKernel.cpp index b2e44f8e09..fcd1594601 100644 --- a/src/core/NEON/kernels/NEWinogradLayerKernel.cpp +++ b/src/core/NEON/kernels/NEWinogradLayerKernel.cpp @@ -299,12 +299,11 @@ void NEWinogradLayerTransformOutputKernel<T, OutputTileRows, OutputTileCols, Ker { ARM_COMPUTE_UNUSED(info); ARM_COMPUTE_ERROR_ON_UNCONFIGURED_KERNEL(this); - ARM_COMPUTE_ERROR_ON_NULLPTR(_biases->buffer()); ARM_COMPUTE_ERROR_ON_NULLPTR(_output_workspace); ARM_COMPUTE_ERROR_ON_NULLPTR(_output); OutputTransform output_transform(_output_workspace, _matrix_stride, _matrix_row_stride, - reinterpret_cast<T *>(_biases->buffer()), _output, + (_biases ? reinterpret_cast<T *>(_biases->buffer()) : nullptr), _output, _n_batches, _n_rows, _n_cols, _n_channels); // The code below cannot be moved to configure because biases hasn't been allocated at that point |