diff options
Diffstat (limited to 'arm_compute/core/NEON/kernels/NEDepthToSpaceLayerKernel.h')
-rw-r--r-- | arm_compute/core/NEON/kernels/NEDepthToSpaceLayerKernel.h | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/arm_compute/core/NEON/kernels/NEDepthToSpaceLayerKernel.h b/arm_compute/core/NEON/kernels/NEDepthToSpaceLayerKernel.h index 3129ea9fb9..0b645887ee 100644 --- a/arm_compute/core/NEON/kernels/NEDepthToSpaceLayerKernel.h +++ b/arm_compute/core/NEON/kernels/NEDepthToSpaceLayerKernel.h @@ -75,6 +75,7 @@ private: const ITensor *_input; /**< Source tensor */ ITensor *_output; /**< Destination tensor */ int32_t _block_shape; /**< Block shape */ + DataLayout _data_layout; /**< Data layout of the operation */ }; } // namespace arm_compute #endif /* ARM_COMPUTE_NEDEPTHTOSPACELAYERKERNEL_H */ |