flax
92b0cdd4 - Flax layers are parametrizable with custom conv_general_dilated.

Commit

2 years ago

Flax layers are parametrizable with custom conv_general_dilated. PiperOrigin-RevId: 509679602

Author

a-googler

Committer

a-googler

Parents