diff options
| author | Georgios Pinitas <georgios.pinitas@arm.com> | 2018-11-16 18:19:43 +0000 |
|---|---|---|
| committer | Georgios Pinitas <georgios.pinitas@arm.com> | 2018-11-20 18:45:09 +0000 |
| commit | c47ef20d69e8ea0f519fdc679435cd7037fc18fe (patch) | |
| tree | 4f3aaeb62a5bc2d6177d2212a1322b8eb17e49fc /arm_compute/runtime/NEON/functions/NEScale.h | |
| parent | 2a35f425dc65afb517bba396e1c3be98ec7f507f (diff) | |
| download | ComputeLibrary-c47ef20d69e8ea0f519fdc679435cd7037fc18fe.tar.gz | |
COMPMID-1646: NEResizeBilinearLayer NHWC
-Adds NHWC support for FP16
Change-Id: I61addf8efecf511ac8cd5f8aa9afc3e09c476aaf
Diffstat (limited to 'arm_compute/runtime/NEON/functions/NEScale.h')
| -rw-r--r-- | arm_compute/runtime/NEON/functions/NEScale.h | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/arm_compute/runtime/NEON/functions/NEScale.h b/arm_compute/runtime/NEON/functions/NEScale.h index 9b5a12bece..d59e3cccb6 100644 --- a/arm_compute/runtime/NEON/functions/NEScale.h +++ b/arm_compute/runtime/NEON/functions/NEScale.h @@ -47,7 +47,7 @@ public: NEScale(); /** Initialize the function's source, destination, interpolation type and border_mode. * - * @param[in, out] input Source tensor. Data type supported: U8/S16/F32. (Written to only for @p border_mode != UNDEFINED) + * @param[in, out] input Source tensor. Data type supported: U8/S16/F16/F32. (Written to only for @p border_mode != UNDEFINED) * @param[out] output Destination tensor. Data type supported: Same as @p input. All but the lowest two dimensions must be the same size as in the input tensor, i.e. scaling is only performed within the XY-plane. * @param[in] policy The interpolation type. * @param[in] border_mode Strategy to use for borders. @@ -58,7 +58,7 @@ public: SamplingPolicy sampling_policy = SamplingPolicy::CENTER); /** Static function to check if given info will lead to a valid configuration of @ref NEScale * - * @param[in] input Source tensor. Data type supported: U8/S16/F32. (Written to only for @p border_mode != UNDEFINED) + * @param[in] input Source tensor. Data type supported: U8/S16/F16/F32. (Written to only for @p border_mode != UNDEFINED) * @param[in] output Destination tensor. Data type supported: Same as @p input. All but the lowest two dimensions must be the same size as in the input tensor, i.e. scaling is only performed within the XY-plane. * @param[in] policy The interpolation type. * @param[in] border_mode Strategy to use for borders. |