How to use the eagerpy.tensor.tensorflow function in eagerpy
To help you get started, we’ve selected a few eagerpy examples, based on popular ways it is used in public projects.
Secure your code as it's written. Use Snyk Code to scan source code in minutes - no build needed - and fix issues immediately.
jonasrauber / eagerpy / eagerpy / tensor / tensorflow.py View on Github 
def wrapper(self, *args, **kwargs):
with tf.device(self.tensor.device):
out = f(self, *args, **kwargs)
return outeagerpy
EagerPy is a thin wrapper around PyTorch, TensorFlow Eager, JAX and NumPy that unifies their interface and thus allows writing code that works natively across all of them.
