flax
Add broadcast of `strides` and `kernel_dilation` to `nn.ConvTranspose`
#3731

Merged

Add broadcast of `strides` and `kernel_dilation` to `nn.ConvTranspose` #3731

copybara-service merged 1 commit into google:main from IvyZX:conv-trans

Add broadcast of strides and kernel_dilation to ConvTranspose

fdbc640c

IvyZX requested a review from

chiamp 2 years ago

IvyZX requested a review from

levskaya 2 years ago

levskaya approved these changes on 2024-03-01

IvyZX assigned

IvyZX 2 years ago

IvyZX added pull ready

copybara-service merged 1abfa872 into main 2 years ago

Reviewers

levskaya

chiamp

Assignees

IvyZX

Labels

pull ready

Milestone

No milestone