diff options
Diffstat (limited to 'src')
-rw-r--r-- | src/backends/cl/workloads/ClSoftmaxUint8Workload.cpp | 7 | ||||
-rw-r--r-- | src/backends/neon/workloads/NeonSoftmaxUint8Workload.cpp | 7 |
2 files changed, 6 insertions, 8 deletions
diff --git a/src/backends/cl/workloads/ClSoftmaxUint8Workload.cpp b/src/backends/cl/workloads/ClSoftmaxUint8Workload.cpp index 84d735cdf7..ce2a9e6356 100644 --- a/src/backends/cl/workloads/ClSoftmaxUint8Workload.cpp +++ b/src/backends/cl/workloads/ClSoftmaxUint8Workload.cpp @@ -25,10 +25,9 @@ ClSoftmaxUint8Workload::ClSoftmaxUint8Workload(const SoftmaxQueueDescriptor& des const auto outputQuantization = output.info()->quantization_info(); - if (((!outputQuantization.scale.empty()) && (outputQuantization.scale[0] != (1.0f / 256.0f))) || - ((!outputQuantization.offset.empty()) && (outputQuantization.offset[0] != 0)) || - (outputQuantization.scale.empty()) || - (outputQuantization.offset.empty())) + if ((!outputQuantization.scale().empty() && outputQuantization.scale()[0] != (1.0f / 256.0f)) || + (!outputQuantization.offset().empty() && outputQuantization.offset()[0] != 0) || + outputQuantization.scale().empty() || outputQuantization.offset().empty()) { throw InvalidArgumentException( "Invalid quantization for output. Only scale = 1.0f / 256.0f and offset = 0 supported"); diff --git a/src/backends/neon/workloads/NeonSoftmaxUint8Workload.cpp b/src/backends/neon/workloads/NeonSoftmaxUint8Workload.cpp index d1e49d954c..363c1502f5 100644 --- a/src/backends/neon/workloads/NeonSoftmaxUint8Workload.cpp +++ b/src/backends/neon/workloads/NeonSoftmaxUint8Workload.cpp @@ -25,10 +25,9 @@ NeonSoftmaxUint8Workload::NeonSoftmaxUint8Workload(const SoftmaxQueueDescriptor& const auto outputQuantization = output.info()->quantization_info(); - if (((!outputQuantization.scale.empty()) && (outputQuantization.scale[0] != (1.0f / 256.0f))) || - ((!outputQuantization.offset.empty()) && (outputQuantization.offset[0] != 0)) || - (outputQuantization.scale.empty()) || - (outputQuantization.offset.empty())) + if ((!outputQuantization.scale().empty() && outputQuantization.scale()[0] != (1.0f / 256.0f)) || + (!outputQuantization.offset().empty() && outputQuantization.offset()[0] != 0) || + outputQuantization.scale().empty() || outputQuantization.offset().empty()) { throw InvalidArgumentException( "Invalid quantization for output. Only scale = 1.0f / 256.0f and offset = 0 supported"); |