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

Fixed bug for pad_v2 

Signed-off-by: Chen Xin <jack.chen@verisilicon.com>
This commit is contained in:
Chen Xin 2022-12-27 16:57:26 +08:00 committed by Sven
parent 0e211c8efd
commit c1f8a959af
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
src/tim/transform/ops/pad_v2_layout_inference.h
View File

@ -44,14 +44,14 @@ class PadV2LayoutInfer : public OpLayoutInfer {
auto i_src = op_->impl()->InputsTensor()[0]; auto i_src = op_->impl()->InputsTensor()[0];
auto input_pv = context_->GetPermuteVector(i_src); 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<uint32_t> front_size(dim_num); std::vector<uint32_t> front_size(dim_num);
std::vector<uint32_t> back_size(dim_num); std::vector<uint32_t> 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); 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); 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()) { if (!input_pv->IsAligned()) {
front_size = MapMultipleAxis(input_pv->AsStdVec(), front_size); front_size = MapMultipleAxis(input_pv->AsStdVec(), front_size);