Skip to content

Commit

Permalink
Creation of a ConvND class. More flexible masks. Lots more tests.
Browse files Browse the repository at this point in the history 
Creating a ConvND class that is then subclassed for Conv{1,2,3}D. This removes a lot of duplicated boilerplate code. As part of this we, are also adding a `mask` argument to the Conv{1,3}D classes. Finally, we remove rank restrictions on the mask argument. We add testing for new masking functionality.

PiperOrigin-RevId: 179918026
  • Loading branch information
@diegolascasas
Deepmind authored and diegolascasas committed Jan 8, 2018
1 parent d0f50a6 commit ca13720
Show file tree
Hide file tree
Showing 2 changed files with 615 additions and 580 deletions.

0 comments on commit ca13720

Please sign in to comment.