diff --git a/mindspore/ccsrc/minddata/dataset/kernels/image/lite_cv/image_process.cc b/mindspore/ccsrc/minddata/dataset/kernels/image/lite_cv/image_process.cc index 00e670bd91..a296445611 100644 --- a/mindspore/ccsrc/minddata/dataset/kernels/image/lite_cv/image_process.cc +++ b/mindspore/ccsrc/minddata/dataset/kernels/image/lite_cv/image_process.cc @@ -973,7 +973,9 @@ bool Subtract(const LiteMat &src1, const LiteMat &src2, LiteMat &dst) { size_t total_size = src1.height_ * src1.width_ * src1.channel_; - if (src1.data_type_ == LDataType::INT8) { + if (src1.data_type_ == LDataType::BOOL) { + SubtractImpl(src1, src2, dst, total_size); + } else if (src1.data_type_ == LDataType::INT8) { SubtractImpl(src1, src2, dst, total_size); } else if (src1.data_type_ == LDataType::UINT8) { SubtractImpl(src1, src2, dst, total_size); @@ -1053,9 +1055,7 @@ bool Divide(const LiteMat &src1, const LiteMat &src2, LiteMat &dst) { size_t total_size = src1.height_ * src1.width_ * src1.channel_; - if (src1.data_type_ == LDataType::BOOL) { - DivideImpl(src1, src2, dst, total_size); - } else if (src1.data_type_ == LDataType::INT8) { + if (src1.data_type_ == LDataType::INT8) { DivideImpl(src1, src2, dst, total_size); } else if (src1.data_type_ == LDataType::UINT8) { DivideImpl(src1, src2, dst, total_size);