From 48a60f9f7b0b7b5cf38253b7a2ac576aac43ef78 Mon Sep 17 00:00:00 2001 From: Pablo Palmier Date: Wed, 18 Oct 2017 11:03:08 +0100 Subject: IVGCVSW-632 CL support for Softmax beta parameter Change-Id: I21da48d2f40aa900301235eaced54b7eb644b0b2 Reviewed-on: http://mpd-gerrit.cambridge.arm.com/91307 Tested-by: Kaizen Reviewed-by: Anthony Barbier --- src/runtime/CL/functions/CLSoftmaxLayer.cpp | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'src/runtime/CL/functions/CLSoftmaxLayer.cpp') diff --git a/src/runtime/CL/functions/CLSoftmaxLayer.cpp b/src/runtime/CL/functions/CLSoftmaxLayer.cpp index 7505a2c974..fa324ee61d 100644 --- a/src/runtime/CL/functions/CLSoftmaxLayer.cpp +++ b/src/runtime/CL/functions/CLSoftmaxLayer.cpp @@ -35,7 +35,7 @@ CLSoftmaxLayer::CLSoftmaxLayer(std::shared_ptr memory_manager) { } -void CLSoftmaxLayer::configure(const ICLTensor *input, ICLTensor *output) +void CLSoftmaxLayer::configure(const ICLTensor *input, ICLTensor *output, float beta) { ARM_COMPUTE_ERROR_ON_DATA_TYPE_CHANNEL_NOT_IN(input, 1, DataType::QS8, DataType::QS16, DataType::F16, DataType::F32); @@ -55,7 +55,7 @@ void CLSoftmaxLayer::configure(const ICLTensor *input, ICLTensor *output) // Configure Kernels _max_kernel.configure(input, &_max); - _shift_exp_sum_kernel.configure(input, &_max, &_tmp, &_sum); + _shift_exp_sum_kernel.configure(input, &_max, &_tmp, &_sum, beta); _norm_kernel.configure(&_tmp, &_sum, output); // Allocate intermediate buffers -- cgit v1.2.1