diff options
| author | Georgios Pinitas <georgios.pinitas@arm.com> | 2019-12-20 10:29:12 +0000 |
|---|---|---|
| committer | Georgios Pinitas <georgios.pinitas@arm.com> | 2019-12-20 11:54:26 +0000 |
| commit | 0e4556ce45cb857ac44354df73f8228475f3ad3c (patch) | |
| tree | 4119efa5206f58c1ceb31b6d0dd53f1484ab352f /arm_compute/core/NEON/kernels/NEDepthToSpaceLayerKernel.h | |
| parent | 0779fecbf897fe85c5e13da52b129e439c4cc75d (diff) | |
| download | ComputeLibrary-0e4556ce45cb857ac44354df73f8228475f3ad3c.tar.gz | |
COMPMID-2819: Retain data layout in NESpaceToDepth and NEDepthToSpace
Retain layout from configuration to execution as following configuration
might alter tensor meta-data in client code.
Signed-off-by: Georgios Pinitas <georgios.pinitas@arm.com>
Change-Id: Ifdbb285809956b966ea534ab3ab231911a307cc0
Reviewed-on: https://review.mlplatform.org/c/2512
Reviewed-by: Pablo Marquez <pablo.tello@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
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 */ |