Skip to content

Conversation

@Zheng-Bicheng
Copy link
Contributor

@Zheng-Bicheng Zheng-Bicheng commented Feb 21, 2024

Support conv2d when data_format is NHWC

@Zheng-Bicheng
Copy link
Contributor Author

@jiangjiajun 我添加了对数据类型为NHWC的conv2d算子的支持

@Zheng-Bicheng
Copy link
Contributor Author

@Liliyaw 更新完FP32再更新一个int8应该就可以在ethos-u上运行了

@Zheng-Bicheng
Copy link
Contributor Author

@jiangjiajun 我尝试对data_format为"NHWC"的conv2d算子添加单测,但是由于PaddleInference目前对NHWC算子的推理存在Bug(详情见 PaddlePaddle Issues 61234),在不破坏当前单测结构的情况下,可能需要延后添加单测。

Copy link
Contributor

@jiangjiajun jiangjiajun left a comment

Choose a reason for hiding this comment

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

LGTM

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