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.

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

(allenl): Consider another way to gather device information. Lower #4330

Open
piascikj opened this issue Apr 26, 2018 · 0 comments
Open

(allenl): Consider another way to gather device information. Lower #4330

piascikj opened this issue Apr 26, 2018 · 0 comments
Labels

Comments

@piascikj
Copy link
Member

priority since this property isn't part of the normal save()/restore()
workflow, but does come up when some alternative builders are passed to
the Saver.


opened via imdone.io from a code comment on 236120d32d1c720ff72f617792d268ec2c82d9e6 by Allen Lavoie [email protected]


# TODO(allenl): Consider another way to gather device information. Lower
# priority since this property isn't part of the normal save()/restore()
# workflow, but does come up when some alternative builders are passed to
# the Saver.
if self._device is None:
self._device = self.specs[0].tensor.device
return self._device
def restore(self, restored_tensors, restored_shapes):

@piascikj piascikj added the TODO label May 2, 2018
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
Projects
None yet
Development

No branches or pull requests

1 participant