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.
I noticed that the method
index_updateseems very slow for tensorflow backend when used in compiled code (a function wrapped withtf.function).It could be worth to see how we could enhance the current implementation.