From 68b5acbe7c90bceae073d4bfca60ffa92dff0fc4 Mon Sep 17 00:00:00 2001 From: Zhouheng Zheng Date: Thu, 4 Nov 2021 19:13:21 +0800 Subject: [PATCH] Fix layout inference bug for resize layer(#205) Co-authored-by: zhouheng.zheng --- src/tim/transform/ops/resize_layout_inference.h | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/tim/transform/ops/resize_layout_inference.h b/src/tim/transform/ops/resize_layout_inference.h index 4653513..530907f 100644 --- a/src/tim/transform/ops/resize_layout_inference.h +++ b/src/tim/transform/ops/resize_layout_inference.h @@ -51,7 +51,7 @@ class ResizeLayoutInfer : public OpLayoutInfer { auto final_pv = input_pv->Reverse()->Add(required_pv); 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_->SetPermuteVector(i_src, final_pv); }