AccelerateHS / AccelerateHS/accelerate

Feature: Strided convolutions

Aperta
#387 5 commenti 0 reazioni 0 assegnatari Vedi su GitHub
language new feature
Lingua principale
Haskell
Stelle
1k
Fork
135
Metriche di merge delle PR
Nessuna PR unita negli ultimi 30g

Descrizione

Something like this would be amazing (I can't figure out how to do it efficiently in accelerate proper)
``` haskell
stridedConvolution
:: Exp sh --stride
-> Exp sh --padding?
-> (Exp a -> Exp b -> Exp c) --multiply
-> (Exp c -> Exp c -> Exp d) --add
-> Array sh b --kernel
-> Array sh a --source
-> Array sh d
```

Guida per i contributori

Nessuna guida per i contributori indicizzata per questo repository

Valutazione

Questa issue non è ancora stata valutata.

Ricevi le nuove issue nella tua casella

Un breve riepilogo di issue GitHub adatte ai principianti.