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

zeros creates constant locally #477

Open
Keno opened this issue Dec 14, 2018 · 2 comments
Open

zeros creates constant locally #477

Keno opened this issue Dec 14, 2018 · 2 comments

Comments

@Keno
Copy link
Contributor

Keno commented Dec 14, 2018

The TF.jl implementation of zeros creates an array of zeros locally and then puts it into the graph. This is problematic if you're using a small client to drive a large distributed cluster (e.g. a TPU), both because the client can run out of memory and because you run into max message size limits.
@Keno
Copy link
Contributor Author

Keno commented Dec 17, 2018

FWIW, fill(TensorFlow.constant(0f0), dims...) seems to work fairly well as an implementation here.

@malmaud
Copy link
Owner

malmaud commented Dec 19, 2018

Wanna just submit a PR to do that?

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
2 participants
@malmaud @Keno