Skip to content

[OpenCL]fix tf_shufflenet model segmentation bug#5777

Merged
daming5432 merged 2 commits intoPaddlePaddle:developfrom
daming5432:fix_tf_shufflenet
Mar 23, 2021
Merged

[OpenCL]fix tf_shufflenet model segmentation bug#5777
daming5432 merged 2 commits intoPaddlePaddle:developfrom
daming5432:fix_tf_shufflenet

Conversation

@daming5432
Copy link
Collaborator

导致tf_shufflenet segmentation的原因是原先的layout转换中不支持5维的tensor,本pr添加layout对5维tensor的支持。

@paddle-bot-old
Copy link

Thanks for your contribution!

ysh329
ysh329 previously approved these changes Mar 23, 2021
Copy link
Contributor

@ysh329 ysh329 left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM,考虑是否需要增加对>=6维度的LOG(FATAL)

Copy link
Contributor

@ysh329 ysh329 left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM

@daming5432 daming5432 merged commit 9b0436c into PaddlePaddle:develop Mar 23, 2021
@daming5432 daming5432 deleted the fix_tf_shufflenet branch March 23, 2021 10:13
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants