search

jonasrauber / eagerpy

PyTorch, TensorFlow, JAX and NumPy — all of them natively using the same code
https://eagerpy.jonasrauber.de
MIT License
693 stars 39 forks source link

`index_update` seems very slow for tensorflow backend #42

Open eserie opened 3 years ago

eserie commented 3 years ago

I noticed that the method index_update seems very slow for tensorflow backend when used in compiled code (a function wrapped with tf.function).

It could be worth to see how we could enhance the current implementation.