Closed madrugado closed 7 years ago
@madrugado There's a merge conflict in this PR.
I have tested this on a project derived from this code and it works perfectly fine.
@madrugado @hugovk @wichtounet Thanks guys, I also tested and training runs on my GPU just fine without the explicit device call.
TF could place operations to appropriate devices without setting this explicitly, the embedding op is implemented on GPU since 0.9, so the declaration could be safely omitted. Also the placement on GPU gives performance boost about 5-10% in my experiments