diff options
Diffstat (limited to 'src')
-rw-r--r-- | src/cpu/kernels/CpuActivationKernel.cpp | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/src/cpu/kernels/CpuActivationKernel.cpp b/src/cpu/kernels/CpuActivationKernel.cpp index 7d13817eab..2ff54cd6e7 100644 --- a/src/cpu/kernels/CpuActivationKernel.cpp +++ b/src/cpu/kernels/CpuActivationKernel.cpp @@ -190,7 +190,7 @@ void CpuActivationKernel::configure(const ITensorInfo *src, ITensorInfo *dst, Ac if(activation_info.activation() == ActivationLayerInfo::ActivationFunction::HARD_SWISH && src->data_type() == DataType::QASYMM8) { - activation_info.init_lut(src->quantization_info().uniform(), dst->quantization_info().uniform()); + activation_info.init_lut(src->quantization_info().uniform(),(dst)?dst->quantization_info().uniform():src->quantization_info().uniform()); } _act_info = activation_info; |