From 6c9fb23d815c4cb92b087e485b00d62c27cbb27c Mon Sep 17 00:00:00 2001 From: jiangzhiwen Date: Fri, 30 Oct 2020 15:44:28 +0800 Subject: [PATCH] remove bool division --- .../dataset/kernels/image/lite_cv/image_process.cc | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) 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);