colin
/
TIM-VX
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix layout inference bug for resize layer(#205) 

Co-authored-by: zhouheng.zheng <zhouheng.zheng@ouotlook.com>
This commit is contained in:
Zhouheng Zheng 2021-11-04 19:13:21 +08:00 committed by GitHub
parent c9086e0afe
commit 68b5acbe7c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/tim/transform/ops/resize_layout_inference.h
View File

@ -51,7 +51,7 @@ class ResizeLayoutInfer : public OpLayoutInfer {
auto final_pv = input_pv->Reverse()->Add(required_pv); auto final_pv = input_pv->Reverse()->Add(required_pv);
if (!final_pv->IsAligned()) { if (!final_pv->IsAligned()) {
auto perm_out = InsertPermute(i_src, final_pv); auto perm_out = InsertPermute(context_->GetMapedTensor(i_src), final_pv);
context_->UpdateTensorMap(i_src, perm_out); context_->UpdateTensorMap(i_src, perm_out);
context_->SetPermuteVector(i_src, final_pv); context_->SetPermuteVector(i_src, final_pv);
} }