1 files changed, 63 insertions, 0 deletions
diff --git a/libs/ARMComputeEx/arm_compute/runtime/CL/functions/CLActivationLayerEx.h b/libs/ARMComputeEx/arm_compute/runtime/CL/functions/CLActivationLayerEx.h
new file mode 100644
index 000000000..7e578550f
--- /dev/null
+++ b/libs/ARMComputeEx/arm_compute/runtime/CL/functions/CLActivationLayerEx.h
@@ -0,0 +1,63 @@
+/*
+ * Copyright (c) 2018 Samsung Electronics Co., Ltd. All Rights Reserved
+ * Copyright (c) 2016-2018 ARM Limited.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *      http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+#ifndef __ARM_COMPUTE_CLACTIVATIONLAYEREX_H__
+#define __ARM_COMPUTE_CLACTIVATIONLAYEREX_H__
+
+#include "arm_compute/runtime/CL/ICLSimpleFunction.h"
+
+#include "arm_compute/core/TypesEx.h"
+
+namespace arm_compute
+{
+class ICLTensor;
+
+/** Basic function to run @ref CLActivationLayerExKernel
+ *
+ * @note The function simulates an activation layer with the specified activation function.
+ */
+class CLActivationLayerEx : public ICLSimpleFunction
+{
+public:
+  /** Set the input and output tensor.
+   *
+   * @note If the output tensor is a nullptr or is equal to the input, the activation function will
+   * be performed in-place
+   *
+   * @param[in, out] input    Source tensor. In case of @p output tensor = nullptr, this tensor will
+   * store the result
+   *                          of the activation function. Data types supported:
+   * QASYMM8/F16/F32.
+   * @param[out]     output   Destination tensor. Data type supported: same as @p input
+   * @param[in]      act_info Activation layer parameters.
+   */
+  void configure(ICLTensor *input, ICLTensor *output, ActivationLayerInfoEx act_info);
+  /** Static function to check if given info will lead to a valid configuration of @ref
+   * CLActivationLayer
+   *
+   * @param[in] input    Source tensor info. In case of @p output tensor info = nullptr, this tensor
+   * will store the result
+   *                     of the activation function. Data types supported: QASYMM8/F16/F32.
+   * @param[in] output   Destination tensor info. Data type supported: same as @p input
+   * @param[in] act_info Activation layer information.
+   *
+   * @return a status
+   */
+  static Status validate(const ITensorInfo *input, const ITensorInfo *output,
+                         const ActivationLayerInfoEx &act_info);
+};
+}
+#endif /* __ARM_COMPUTE_CLACTIVATIONLAYEREX_H__ */