This website works better with JavaScript.
Home
Issues
Pull Requests
Milestones
AI流水线
Repositories
Datasets
Forum
实训
竞赛
大数据
AI开发
Register
Sign In
Huawei_Technology
/
mindspore
Not watched
Unwatch
Watch all
Watch but not notify
1
Star
0
Fork
0
Code
Releases
13
Wiki
evaluate
Activity
Issues
0
Pull Requests
0
Datasets
Model
Cloudbrain
HPC
Browse Source
fix resize bug
tags/v1.1.0
gongdaguo
5 years ago
parent
c624b04b24
commit
d18a8cbb9c
1 changed files
with
4 additions
and
0 deletions
Split View
Diff Options
Show Stats
Download Patch File
Download Diff File
+4
-0
mindspore/lite/src/runtime/kernel/arm/fp32/resize_fp32.cc
+ 4
- 0
mindspore/lite/src/runtime/kernel/arm/fp32/resize_fp32.cc
View File
@@ -42,6 +42,10 @@ int ResizeCPUKernel::Init() {
int ResizeCPUKernel::ReSize() {
int ret = RET_OK;
if (method_ == static_cast<int>(schema::ResizeMethod_LINEAR)) {
if (!const_shape_) {
new_height_ = out_tensors_.at(0)->shape()[1];
new_width_ = out_tensors_.at(0)->shape()[2];
}
FreeTmpBuffer();
ret = MallocTmpBuffer();
if (ret != RET_OK) {
Write
Preview
Loading…
Cancel
Save