diff options
Diffstat (limited to 'arm_compute/runtime/NEON/functions/NEFillBorder.h')
| -rw-r--r-- | arm_compute/runtime/NEON/functions/NEFillBorder.h | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/arm_compute/runtime/NEON/functions/NEFillBorder.h b/arm_compute/runtime/NEON/functions/NEFillBorder.h index b8266d9d67..0ae04cbf00 100644 --- a/arm_compute/runtime/NEON/functions/NEFillBorder.h +++ b/arm_compute/runtime/NEON/functions/NEFillBorder.h @@ -31,6 +31,7 @@ namespace arm_compute { +// Forward declaration class ITensor; /** Basic function to run @ref NEFillBorderKernel */ @@ -41,7 +42,7 @@ public: * * @note This function fills the borders within the XY-planes. * - * @param[in, out] input Source tensor. Data type supported: U8/S16/S32/F32 + * @param[in, out] input Source tensor. Data type supported: All * @param[in] border_width Width of the tensor border in pixels. * @param[in] border_mode Strategy to use for borders. * @param[in] constant_border_value (Optional) Constant value to use for borders if border_mode is set to CONSTANT. @@ -54,5 +55,5 @@ public: private: NEFillBorderKernel _border_handler; /**< Kernel to handle image borders */ }; -} +} // namespace arm_compute #endif /*ARM_COMPUTE_NEFILLBORDER_H */ |