From c1f8a959af4879f74653b9b61c7864b072aacb74 Mon Sep 17 00:00:00 2001 From: Chen Xin Date: Tue, 27 Dec 2022 16:57:26 +0800 Subject: [PATCH] Fixed bug for pad_v2 Signed-off-by: Chen Xin --- src/tim/transform/ops/pad_v2_layout_inference.h | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/src/tim/transform/ops/pad_v2_layout_inference.h b/src/tim/transform/ops/pad_v2_layout_inference.h index 595b44e..7a0c067 100644 --- a/src/tim/transform/ops/pad_v2_layout_inference.h +++ b/src/tim/transform/ops/pad_v2_layout_inference.h @@ -44,14 +44,14 @@ class PadV2LayoutInfer : public OpLayoutInfer { auto i_src = op_->impl()->InputsTensor()[0]; auto input_pv = context_->GetPermuteVector(i_src); - uint32_t dim_num = op_->impl()->node()->nn_param.pad.dim_num; + uint32_t dim_num = op_->impl()->node()->nn_param.pad2.dim_num; std::vector front_size(dim_num); std::vector back_size(dim_num); - memcpy(front_size.data(), op_->impl()->node()->nn_param.pad.front_size, + memcpy(front_size.data(), op_->impl()->node()->nn_param.pad2.front_size, sizeof(uint32_t) * dim_num); - memcpy(back_size.data(), op_->impl()->node()->nn_param.pad.back_size, + memcpy(back_size.data(), op_->impl()->node()->nn_param.pad2.back_size, sizeof(uint32_t) * dim_num); - float pad_value = op_->impl()->node()->nn_param.pad.const_val; + float pad_value = op_->impl()->node()->nn_param.pad2.const_val; if (!input_pv->IsAligned()) { front_size = MapMultipleAxis(input_pv->AsStdVec(), front_size);