SylorHuang
/
mindspore
forked from OSSInnovation/mindspore
0
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

!3884 fix space to depth bug 

Merge pull request !3884 from zhongligeng/master
This commit is contained in:
mindspore-ci-bot 2020-08-03 21:49:16 +08:00 committed by Gitee
parent aa65cbf733 58e6227a23
commit 0f4f343e6b
3 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mindspore/lite/src/runtime/kernel/arm/fp32/space_to_depth.cc
View File

@ -77,6 +77,8 @@ int SpaceToDepthRun(int task_id, LiteParallelGroupEnv *penv, void *cdata) {
}
int SpaceToDepthCPUKernel::Run() {
input_ptr_ = reinterpret_cast<float *>(inputs_[0]->Data());
output_ptr_ = reinterpret_cast<float *>(outputs_[0]->Data());
if (inputs_[0]->GetFormat() == schema::Format_NHWC) {
int ret = LiteBackendParallelLaunch(SpaceToDepthRun, this, thread_h_num_);
if (ret != RET_OK) {

2
mindspore/lite/test/ut/src/runtime/kernel/arm/fp32/space_to_batch_fp32_tests.cc
View File

@ -158,6 +158,8 @@ TEST_F(SpaceToBatchTestFp32, SpaceToBatchTest3) {
}
std::cout << "\n";
CompareOutputData(output.data(), expect_out, out_size, 0.000001);
input_tensor.SetData(nullptr);
output_tensor.SetData(nullptr);
}
} // namespace mindspore

4
mindspore/lite/test/ut/src/runtime/kernel/arm/fp32/space_to_depth_fp32_tests.cc
View File

@ -56,7 +56,7 @@ TEST_F(SpaceToDepthTestFp32, SpaceToDepthTest2) {
input_tensor.SetFormat(schema::Format_NHWC);
input_tensor.set_data_type(kNumberTypeFloat32);
std::vector<lite::tensor::Tensor *> inputs_tensor;
inputs_tensor.emplace_back(&input_tensor);
inputs_tensor.push_back(&input_tensor);
const int out_size = 16;
float expect_out[16] = {1, 2, 10, 20, 5, 6, 3, 8, 18, 10, 11, 55, 3, 4, 15, 25};
@ -68,7 +68,7 @@ TEST_F(SpaceToDepthTestFp32, SpaceToDepthTest2) {
output_tensor.SetFormat(schema::Format_NHWC);
output_tensor.set_data_type(kNumberTypeFloat32);
std::vector<lite::tensor::Tensor *> outputs_tensor;
outputs_tensor.emplace_back(&output_tensor);
outputs_tensor.push_back(&output_tensor);
SpaceToDepthParameter op_param;
op_param.op_parameter_.type_ = schema::PrimitiveType_SpaceToDepth;