This website works better with JavaScript.
Home
Issues
Pull Requests
Milestones
AI流水线
Repositories
Datasets
Forum
实训
竞赛
大数据
AI开发
Register
Sign In
zzy34407230
/
mindspore2022
Not watched
Unwatch
Watch all
Watch but not notify
1
Star
0
Fork
0
Code
Releases
22
Wiki
evaluate
Activity
Issues
0
Pull Requests
0
Datasets
Model
Cloudbrain
HPC
Browse Source
fix lite train resize issue
feature/build-system-rewrite
zhengjun10
4 years ago
parent
b90cf43562
commit
4855e427d6
2 changed files
with
6 additions
and
1 deletions
Split View
Diff Options
Show Stats
Download Patch File
Download Diff File
+5
-0
mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/infer/common_infer.c
+1
-1
mindspore/lite/test/config/models_ms_train.cfg
+ 5
- 0
mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/infer/common_infer.c
View File
@@ -385,6 +385,11 @@ int CommonGradInferShape(const TensorC *const *inputs, size_t inputs_size, Tenso
return NNACL_INFER_INVALID;
}
MS_CHECK_TRUE_RET(inputs[0]->shape_size_ == inputs[1]->shape_size_, NNACL_ERR);
for (int i = 0; i < inputs[0]->shape_size_; i++) {
if (inputs[0]->shape_[i] != inputs[1]->shape_[i]) {
return NNACL_ERR;
}
}
SetShapeTensor(outputs[0], inputs[0]);
return NNACL_OK;
}
+ 1
- 1
mindspore/lite/test/config/models_ms_train.cfg
View File
@@ -45,4 +45,4 @@ unified_api code_example
train_lenet code_example
train_lenet_java code_example
# LAST
test_resize inputShapes 16,10,10,1:16,10,10,1 0.5
#
test_resize inputShapes 16,10,10,1:16,10,10,1 0.5
Write
Preview
Loading…
Cancel
Save