diff options
author | Michele Di Giorgio <michele.digiorgio@arm.com> | 2018-08-24 18:28:48 +0100 |
---|---|---|
committer | Anthony Barbier <anthony.barbier@arm.com> | 2018-11-02 16:54:54 +0000 |
commit | 3e570dbdb0cbcbc3314e8f3e4daf2cf385caf325 (patch) | |
tree | 5cb9a4279a4b6b456a53ff8f918bb2aa6cd68dbe /src/runtime/NEON/functions/NEDepthConvertLayer.cpp | |
parent | daa38559bb01eb3c2985f503e7b2b4dc850a34d3 (diff) | |
download | ComputeLibrary-3e570dbdb0cbcbc3314e8f3e4daf2cf385caf325.tar.gz |
COMPMID-1304: NEDepthConvert : Add support for FP32 -> FP16 and FP16 -> FP32 + validate() function
Change-Id: I12e4696a454744f6d493ab3a53520d3acf3a1a26
Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/145719
Tested-by: Jenkins <bsgcomp@arm.com>
Reviewed-by: Georgios Pinitas <georgios.pinitas@arm.com>
Diffstat (limited to 'src/runtime/NEON/functions/NEDepthConvertLayer.cpp')
-rw-r--r-- | src/runtime/NEON/functions/NEDepthConvertLayer.cpp | 9 |
1 files changed, 7 insertions, 2 deletions
diff --git a/src/runtime/NEON/functions/NEDepthConvertLayer.cpp b/src/runtime/NEON/functions/NEDepthConvertLayer.cpp index 9a75404fcd..0041c1f62e 100644 --- a/src/runtime/NEON/functions/NEDepthConvertLayer.cpp +++ b/src/runtime/NEON/functions/NEDepthConvertLayer.cpp @@ -1,5 +1,5 @@ /* - * Copyright (c) 2016, 2017 ARM Limited. + * Copyright (c) 2016-2018 ARM Limited. * * SPDX-License-Identifier: MIT * @@ -30,9 +30,14 @@ using namespace arm_compute; -void NEDepthConvertLayer::configure(ITensor *input, ITensor *output, ConvertPolicy policy, uint32_t shift) +void NEDepthConvertLayer::configure(const ITensor *input, ITensor *output, ConvertPolicy policy, uint32_t shift) { auto k = arm_compute::support::cpp14::make_unique<NEDepthConvertLayerKernel>(); k->configure(input, output, policy, shift); _kernel = std::move(k); } + +Status NEDepthConvertLayer::validate(const ITensorInfo *input, const ITensorInfo *output, ConvertPolicy policy, uint32_t shift) +{ + return NEDepthConvertLayerKernel::validate(input, output, policy, shift); +} |