diff options
| author | giuros01 <giuseppe.rossini@arm.com> | 2018-11-20 18:34:46 +0000 |
|---|---|---|
| committer | Giuseppe Rossini <giuseppe.rossini@arm.com> | 2018-11-30 18:00:25 +0000 |
| commit | 164a2727d3bbce0e575d24b7db787c85e2e2c203 (patch) | |
| tree | 983fc1f519032ac9a056e19f87e32597ca1874a1 /arm_compute/runtime/CL/functions/CLRNNLayer.h | |
| parent | 7930db48e12dd3a14c1971f41f5b83527efea281 (diff) | |
| download | ComputeLibrary-164a2727d3bbce0e575d24b7db787c85e2e2c203.tar.gz | |
COMPMID-1717: CL: Implement Maximum, Minimum, SquaredDifference
Change-Id: Ice653e48211053bd3cd20a693bd76de6b4efc370
Reviewed-on: https://review.mlplatform.org/270
Reviewed-by: Georgios Pinitas <georgios.pinitas@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'arm_compute/runtime/CL/functions/CLRNNLayer.h')
| -rw-r--r-- | arm_compute/runtime/CL/functions/CLRNNLayer.h | 22 |
1 files changed, 11 insertions, 11 deletions
diff --git a/arm_compute/runtime/CL/functions/CLRNNLayer.h b/arm_compute/runtime/CL/functions/CLRNNLayer.h index ab7407dbfc..fc86992bdf 100644 --- a/arm_compute/runtime/CL/functions/CLRNNLayer.h +++ b/arm_compute/runtime/CL/functions/CLRNNLayer.h @@ -25,8 +25,8 @@ #define __ARM_COMPUTE_CLRNN_LAYER_H__ #include "arm_compute/core/CL/kernels/CLActivationLayerKernel.h" -#include "arm_compute/core/CL/kernels/CLArithmeticAdditionKernel.h" #include "arm_compute/core/CL/kernels/CLCopyKernel.h" +#include "arm_compute/core/CL/kernels/CLElementwiseOperationKernel.h" #include "arm_compute/runtime/CL/ICLSimpleFunction.h" #include "arm_compute/runtime/CL/functions/CLFullyConnectedLayer.h" #include "arm_compute/runtime/CL/functions/CLGEMM.h" @@ -72,16 +72,16 @@ public: void prepare() override; private: - CLMemoryGroup _memory_group; - CLGEMM _gemm_state_f; - CLArithmeticAdditionKernel _add_kernel; - CLActivationLayerKernel _activation_kernel; - CLFullyConnectedLayer _fully_connected_kernel; - CLCopyKernel _copy_kernel; - CLTensor _fully_connected_out; - CLTensor _gemm_output; - CLTensor _add_output; - bool _is_prepared; + CLMemoryGroup _memory_group; + CLGEMM _gemm_state_f; + CLSaturatedArithmeticOperationKernel _add_kernel; + CLActivationLayerKernel _activation_kernel; + CLFullyConnectedLayer _fully_connected_kernel; + CLCopyKernel _copy_kernel; + CLTensor _fully_connected_out; + CLTensor _gemm_output; + CLTensor _add_output; + bool _is_prepared; }; } #endif /* __ARM_COMPUTE_CLRNN_LAYER_H__ */ |