ucb-bar/onnxruntime-riscv

Refactoring: Use the inbuilt im2col NHWC?

Ouverte

#26 ouverte le 27 janv. 2021

 (1 commentaire) (0 réaction) (0 personne assignée)C++ (37 forks)auto 404
enhancementgood first issue

Métriques du dépôt

Stars
 (93 étoiles)
Métriques de merge PR
 (Aucune PR mergée en 30 j)

Description

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.

Guide contributeur