ucb-bar/onnxruntime-riscv

Refactoring: Use the inbuilt im2col NHWC?

Aperta

#26 aperta il 27 gen 2021

Vedi su GitHubGuida contributor
 (1 commento) (0 reazioni) (0 assegnatari)C++ (37 fork)auto 404
enhancementgood first issue

Metriche repository

Star
 (93 stelle)
Metriche merge PR
 (Metriche PR in attesa)

Descrizione

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.

Guida contributor

Direzione di ricerca
Esamina la funzione im2col nhwc personalizzata esistente e la im2col integrata in ORT. Poi rifattorizza spostando la chiamata a im2col all'interno del ciclo for che itera sui gruppi, in modo che operi su un gruppo alla volta.
Tech stack
cpp
Dominio
backendapi
Tipo issue
Refactoring
DifficoltàQuanto dovrebbe essere impegnativa per un nuovo contributor.
2
Tempo stimatoTempo stimato per completare e verificare uno scope piccolo.
1-3 ore
Stato attivitàQuanto è probabile che la issue sia ancora attiva e reviewable.
Attiva
ChiarezzaQuanto chiaramente la issue descrive problema, scope e risultato atteso.
Chiara
Prerequisiti
C++ONNXRuntime internals
Adatta ai principiantiQuanto la issue è adatta a contributor alla prima esperienza.
60