root
/
MNN
mirror of https://github.com/alibaba/MNN.git
1
0
Fork 0
Code Issues Actions 1 Packages Projects Releases Wiki Activity

Merge pull request #2262 from zeekimn/dev_fix_hiai 

build(hiai): fix for update hiai to 100.520.020.010
This commit is contained in:
jxt1234 2023-03-25 10:17:48 +08:00 committed by GitHub
parent eea434c02e 12e9f52d38
commit 1b35d5df84
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
source/backend/hiai/execution/NPUDepthToSpace.cpp
View File

@ -39,7 +39,7 @@ ErrorCode NPUDepthToSpace::onResize(const std::vector<Tensor *> &inputs, const s
(*permuteBefore)
.set_input_x(*xOp1.get())
.set_attr_order({0,2,3,1})
.SetAttr("NCHW_to_NHWC", ge::AttrValue::CreateFrom<ge::AttrValue::INT>(1));
.SetAttr("NCHW_to_NHWC", ge::AttrValue::CreateFrom(static_cast<int64_t>(1)));
(*depthToSpace)
.set_input_x(*permuteBefore.get())
@ -49,7 +49,7 @@ ErrorCode NPUDepthToSpace::onResize(const std::vector<Tensor *> &inputs, const s
(*permuteAfter)
.set_input_x(*depthToSpace.get())
.set_attr_order({0,3,1,2})
.SetAttr("NHWC_to_NCHW", ge::AttrValue::CreateFrom<ge::AttrValue::INT>(1));
.SetAttr("NHWC_to_NCHW", ge::AttrValue::CreateFrom(static_cast<int64_t>(1)));
mNpuBackend->setOutputOps(mOp, {permuteBefore, depthToSpace, permuteAfter}, outputs);
return NO_ERROR;