diff options
Diffstat (limited to 'src/cpu/kernels/elementwise_binary/generic/sve/fp32.cpp')
| -rw-r--r-- | src/cpu/kernels/elementwise_binary/generic/sve/fp32.cpp | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/src/cpu/kernels/elementwise_binary/generic/sve/fp32.cpp b/src/cpu/kernels/elementwise_binary/generic/sve/fp32.cpp index 0f80813d15..2b479f76f1 100644 --- a/src/cpu/kernels/elementwise_binary/generic/sve/fp32.cpp +++ b/src/cpu/kernels/elementwise_binary/generic/sve/fp32.cpp @@ -31,7 +31,7 @@ namespace cpu template <ArithmeticOperation op> void sve_fp32_elementwise_binary(const ITensor *in1, const ITensor *in2, ITensor *out, const Window &window) { - return elementwise_arithmetic_op<op, float32_t>(in1, in2, out, window); + return elementwise_arithmetic_op<float32_t>(in1, in2, out, op, window); } template void sve_fp32_elementwise_binary<ArithmeticOperation::ADD>(const ITensor *in1, const ITensor *in2, ITensor *out, const Window &window); @@ -46,7 +46,7 @@ template void sve_fp32_elementwise_binary<ArithmeticOperation::PRELU>(const ITen template <ComparisonOperation op> void sve_fp32_comparison_elementwise_binary(const ITensor *in1, const ITensor *in2, ITensor *out, const Window &window) { - return elementwise_comparison_op<op, float>(in1, in2, out, window); + return elementwise_comparison_op<float>(in1, in2, out, op, window); } template void sve_fp32_comparison_elementwise_binary<ComparisonOperation::Equal>(const ITensor *in1, const ITensor *in2, ITensor *out, const Window &window); template void sve_fp32_comparison_elementwise_binary<ComparisonOperation::NotEqual>(const ITensor *in1, const ITensor *in2, ITensor *out, const Window &window); |