diff options
Diffstat (limited to 'arm_compute/core/NEON/kernels/NESpaceToDepthLayerKernel.h')
| -rw-r--r-- | arm_compute/core/NEON/kernels/NESpaceToDepthLayerKernel.h | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/arm_compute/core/NEON/kernels/NESpaceToDepthLayerKernel.h b/arm_compute/core/NEON/kernels/NESpaceToDepthLayerKernel.h index c9ecdd26f8..8e1fd36e24 100644 --- a/arm_compute/core/NEON/kernels/NESpaceToDepthLayerKernel.h +++ b/arm_compute/core/NEON/kernels/NESpaceToDepthLayerKernel.h @@ -75,6 +75,7 @@ private: const ITensor *_input; /**< Source tensor */ ITensor *_output; /**< Destination tensor */ int32_t _block_shape; /**< Block shape */ + DataLayout _data_layout; }; } // namespace arm_compute #endif /* __ARM_COMPUTE_NESPACETODEPTHLAYERKERNEL_H__ */ |