diff options
Diffstat (limited to 'src/core/NEON/kernels/NEArithmeticSubtractionKernel.cpp')
-rw-r--r-- | src/core/NEON/kernels/NEArithmeticSubtractionKernel.cpp | 10 |
1 files changed, 9 insertions, 1 deletions
diff --git a/src/core/NEON/kernels/NEArithmeticSubtractionKernel.cpp b/src/core/NEON/kernels/NEArithmeticSubtractionKernel.cpp index 7a2601be26..0695c94927 100644 --- a/src/core/NEON/kernels/NEArithmeticSubtractionKernel.cpp +++ b/src/core/NEON/kernels/NEArithmeticSubtractionKernel.cpp @@ -448,7 +448,7 @@ inline std::pair<Status, Window> validate_and_configure_window(ITensorInfo &inpu { set_format_if_unknown(output, Format::S16); } - else if(input1.data_type() == DataType::F16 && input2.data_type() == DataType::F16) + else if(input1.data_type() == DataType::F16 || input2.data_type() == DataType::F16) { set_format_if_unknown(output, Format::F16); } @@ -456,6 +456,14 @@ inline std::pair<Status, Window> validate_and_configure_window(ITensorInfo &inpu { set_format_if_unknown(output, Format::F32); } + else if(input1.data_type() == DataType::QASYMM8 || input2.data_type() == DataType::QASYMM8) + { + set_data_type_if_unknown(output, DataType::QASYMM8); + } + else if(input1.data_type() == DataType::QASYMM8_SIGNED || input2.data_type() == DataType::QASYMM8_SIGNED) + { + set_data_type_if_unknown(output, DataType::QASYMM8_SIGNED); + } } Window win = calculate_max_window(valid_region, Steps(num_elems_processed_per_iteration)); |