diff options
Diffstat (limited to 'arm_compute/runtime/NEON/functions/NEGEMMConvolutionLayer.h')
-rw-r--r-- | arm_compute/runtime/NEON/functions/NEGEMMConvolutionLayer.h | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/arm_compute/runtime/NEON/functions/NEGEMMConvolutionLayer.h b/arm_compute/runtime/NEON/functions/NEGEMMConvolutionLayer.h index aadc429864..65c2ef7e0b 100644 --- a/arm_compute/runtime/NEON/functions/NEGEMMConvolutionLayer.h +++ b/arm_compute/runtime/NEON/functions/NEGEMMConvolutionLayer.h @@ -275,10 +275,12 @@ private: NEReshapeLayer _reshape_layer; const ITensor *_original_weights; + const ITensor *_original_output; Tensor _im2col_output; Tensor _weights_reshaped; Tensor _gemm_output; + Tensor _gemm_output_3d; Tensor _tmp_output; DataLayout _data_layout; |