Skip to content

Commit

Permalink
Merge #1531
Browse files Browse the repository at this point in the history 
1531: Add grouped convolution r=DhairyaLGandhi a=DhairyaLGandhi

Add support for grouped convolutions. Needed for a bunch of things. Depends on an NNlib branch where the support is present in the Dims Helpers, but needs to be chained through to the conv kernel. Although we now can generate the correct shaped kernels and filters here.

Co-authored-by: Dhairya Gandhi <dhairya@juliacopmuting.com>
Co-authored-by: Dhairya Gandhi <dhairya@juliacomputing.com>
  • Loading branch information
@bors @DhairyaLGandhi
3 people authored Jul 22, 2021
2 parents 9931730 + 98149dd commit eb5077f
Show file tree
Hide file tree
Showing 4 changed files with 177 additions and 133 deletions.
1 change: 0 additions & 1 deletion .gitignore
View file
Original file line number Diff line number Diff line change
Expand Up @@ -6,4 +6,3 @@ docs/build/
docs/site/
deps
.vscode
Manifest.toml
Loading

0 comments on commit eb5077f

Please sign in to comment.