PR #1024 Add axis_name arg to lifted vmap

Add axis_name arg to lifted vmap #1024

copybara-service merged 1 commit into google:master from jheek:vmap-axis_name

Add axis_name arg to lifted vmap

0d2251aa

jheek requested a review from

avital 4 years ago

jheek requested a review from

levskaya 4 years ago

google-cla added cla: yes

avital approved these changes on 2021-02-16

jheek added pull ready

copybara-service merged 33581564 into master 4 years ago

Reviewers

avital

levskaya

Assignees

No one assigned

Labels

cla: yes pull ready

Milestone

No milestone