Browse Source

!21550 [MSLITE] Fix bug of pad operator.

Merge pull request !21550 from wangshaocong/bugfix_master
tags/v1.5.0-rc1
i-robot Gitee 4 years ago
parent
commit
f367d4c9d8
2 changed files with 0 additions and 8 deletions
  1. +0
    -4
      mindspore/lite/src/runtime/kernel/arm/fp16/pad_fp16.cc
  2. +0
    -4
      mindspore/lite/src/runtime/kernel/arm/fp32/pad_fp32.cc

+ 0
- 4
mindspore/lite/src/runtime/kernel/arm/fp16/pad_fp16.cc View File

@@ -27,7 +27,6 @@ using mindspore::schema::PrimitiveType_PadFusion;
namespace mindspore::kernel {
namespace {
constexpr size_t kPadCommonInputSize = 2;
constexpr size_t kPadMaxInputSize = 3;
} // namespace
int PadFp16CPUKernel::RunImpl(int task_id) {
PadFp16(input_, output_, in_, out_, pad_param_->paddings_, task_id, op_parameter_->thread_num_);
@@ -102,9 +101,6 @@ int PadFp16CPUKernel::Run() {
return RET_ERROR;
}
}
if (in_tensors_.size() == kPadMaxInputSize) {
pad_param_->constant_value_ = reinterpret_cast<float *>(in_tensors_.at(2)->data_c())[0];
}
if (pad_param_->constant_value_ - 0.0f < 1e-5) {
memset(output_, 0, output_tensor->ElementsNum() * sizeof(float16_t));
} else {


+ 0
- 4
mindspore/lite/src/runtime/kernel/arm/fp32/pad_fp32.cc View File

@@ -29,7 +29,6 @@ namespace mindspore::kernel {
namespace {
constexpr size_t kMirrorPadInputSize = 2;
constexpr size_t kPadCommonInputSize = 2;
constexpr size_t kPadMaxInputSize = 3;
} // namespace
int PadCPUKernel::Init() {
CHECK_LESS_RETURN(in_tensors_.size(), 1);
@@ -397,9 +396,6 @@ int PadCPUKernel::Run() {
return RET_ERROR;
}
}
if (in_tensors_.size() == kPadMaxInputSize) {
pad_param_->constant_value_ = reinterpret_cast<float *>(in_tensors_.at(2)->data_c())[0];
}
auto output = out_tensors_.at(0);
int output_size = output->ElementsNum();
auto output_data = reinterpret_cast<float *>(output->data_c());


Loading…
Cancel
Save