Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

TF vectorized_map Misses One Positional Argument #5

Open
DarrenZhang01 opened this issue Jul 21, 2020 · 0 comments
Open

TF vectorized_map Misses One Positional Argument #5

DarrenZhang01 opened this issue Jul 21, 2020 · 0 comments
Assignees
@DarrenZhang01
Labels
bug Something isn't working

Comments

@DarrenZhang01
Copy link
Owner

TF vectorized_map requires one more positional argument elems compared to JAX vmap, which stops the running of the helper function _jit_map in test_utils.py. Need a way to fill in this argument and make the code work.
@DarrenZhang01 DarrenZhang01 added the bug Something isn't working label Jul 21, 2020
@DarrenZhang01 DarrenZhang01 self-assigned this Jul 21, 2020
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees

@DarrenZhang01 DarrenZhang01

Labels
bug Something isn't working
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
1 participant
@DarrenZhang01