Browse Source

!8033 [MD] Remove bool division of lite cv

Merge pull request !8033 from jiangzhiwen/fix/remove_bool_divide
tags/v1.1.0
mindspore-ci-bot Gitee 5 years ago
parent
commit
3fda4ce1d1
1 changed files with 4 additions and 4 deletions
  1. +4
    -4
      mindspore/ccsrc/minddata/dataset/kernels/image/lite_cv/image_process.cc

+ 4
- 4
mindspore/ccsrc/minddata/dataset/kernels/image/lite_cv/image_process.cc View File

@@ -974,7 +974,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<bool>(src1, src2, dst, total_size);
} else if (src1.data_type_ == LDataType::INT8) {
SubtractImpl<int8_t>(src1, src2, dst, total_size);
} else if (src1.data_type_ == LDataType::UINT8) {
SubtractImpl<uint8_t>(src1, src2, dst, total_size);
@@ -1054,9 +1056,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<bool>(src1, src2, dst, total_size);
} else if (src1.data_type_ == LDataType::INT8) {
if (src1.data_type_ == LDataType::INT8) {
DivideImpl<int8_t>(src1, src2, dst, total_size);
} else if (src1.data_type_ == LDataType::UINT8) {
DivideImpl<uint8_t>(src1, src2, dst, total_size);


Loading…
Cancel
Save