Does tf.gather work on GPU?

191 views

Skip to first unread message

Anatoly Vostryakov

unread,

Nov 20, 2015, 7:18:51 AM11/20/15

to Discuss

Hi, there!

I discovered that nn.embedding_lookup is equal to tf.gather for a case of one parameter tensor. So far, for nn.embedding_lookup was said that it works only on CPU.

Is it means that tf.gather also works only on CPU?

Reply all

Reply to author

Forward

0 new messages