ucb-bar/onnxruntime-riscv

Refactoring: Use the inbuilt im2col NHWC?

开放

#26 创建于 2021年1月27日

 (1 条评论) (0 个反应) (0 位负责人)C++ (37 个派生)auto 404
enhancementgood first issue

仓库指标

星标
 (93 个星标)
PR 合并指标
 (PR 指标待抓取)

描述

More of a code cleanliness thing, but currently for im2col in nhwc layout we use our own method that I took from the pytorch docs. ONNXRuntime also includes its own im2col nhwc implementation (see the cpu math ops). The two functions are basically identical except in our own im2col_nhwc function it does it for all groups whereas the ORT one only does it for one group. So when you refactor you'll have to move the im2col inside the for-loop where we iterate over the groups.

贡献者指南