projects / platform / upstream / tensorflow.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: 2b932c2) | patch
Extended TFE_OpSetDevice() with the ability to set an op device from non-GPU back...
authorMingsheng Hong <hongm@google.com>
Sat, 3 Feb 2018 03:10:39 +0000 (19:10 -0800)
committerTensorFlower Gardener <gardener@tensorflow.org>
Sat, 3 Feb 2018 03:14:34 +0000 (19:14 -0800)
commit0fab6e888c5f90de3e878566123c1906261ce27e
tree5e44c57525e03c6a4a95ec73303165831e991355tree | snapshot
parent2b932c28d60ec4f248d950cd2ef69a7eb98bb66dcommit | diff
Extended TFE_OpSetDevice() with the ability to set an op device from non-GPU back to GPU.

Added unit testing, and also refined unit test logic for checking the presence
of a GPU device. The latter is needed when we add XLA device support.

PiperOrigin-RevId: 184366172
tensorflow/c/eager/c_api.cc diff | blob | history
tensorflow/c/eager/c_api.h diff | blob | history
tensorflow/c/eager/c_api_test.cc diff | blob | history
Domain: Machine Learning / ML Framework;