diff options
Diffstat (limited to 'src/cpu/kernels/elementwise_binary/generic/sve/fp16.cpp')
| -rw-r--r-- | src/cpu/kernels/elementwise_binary/generic/sve/fp16.cpp | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/src/cpu/kernels/elementwise_binary/generic/sve/fp16.cpp b/src/cpu/kernels/elementwise_binary/generic/sve/fp16.cpp index 8adacbfe67..85224351df 100644 --- a/src/cpu/kernels/elementwise_binary/generic/sve/fp16.cpp +++ b/src/cpu/kernels/elementwise_binary/generic/sve/fp16.cpp @@ -33,7 +33,7 @@ namespace cpu template <ArithmeticOperation op> void sve_fp16_elementwise_binary(const ITensor *in1, const ITensor *in2, ITensor *out, const Window &window) { - return elementwise_arithmetic_op<op, float16_t>(in1, in2, out, window); + return elementwise_arithmetic_op<float16_t>(in1, in2, out, op, window); } template void sve_fp16_elementwise_binary<ArithmeticOperation::ADD>(const ITensor *in1, const ITensor *in2, ITensor *out, const Window &window); @@ -48,7 +48,7 @@ template void sve_fp16_elementwise_binary<ArithmeticOperation::PRELU>(const ITen template <ComparisonOperation op> void sve_fp16_comparison_elementwise_binary(const ITensor *in1, const ITensor *in2, ITensor *out, const Window &window) { - return elementwise_comparison_op<op, float16_t>(in1, in2, out, window); + return elementwise_comparison_op<float16_t>(in1, in2, out, op, window); } template void sve_fp16_comparison_elementwise_binary<ComparisonOperation::Equal>(const ITensor *in1, const ITensor *in2, ITensor *out, const Window &window); |