projects / platform / upstream / tensorflow.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: 4ae7e05) | patch
Skip tests that require unavailable hardware when not using DistributionStrategy
authorIgor Saprykin <isaprykin@google.com>
Fri, 18 May 2018 22:49:01 +0000 (15:49 -0700)
committerTensorFlower Gardener <gardener@tensorflow.org>
Fri, 18 May 2018 22:51:52 +0000 (15:51 -0700)
commit88ec2f68522495d13d8efc5542c7999e841b85e5
treea4d39d658d2c1885fa0019646be61c64e29b98b9tree | snapshot
parent4ae7e05f89b1c88cf6dde955d419aef474ae3ef1commit | diff
Skip tests that require unavailable hardware when not using DistributionStrategy

Right now combinations.py skips tests that do not have the hardware that's
requried by the DistributionStrategy instance that is used in that test.  After
this change, the user can trigger such a behavior even when they are not using
DistributionStrategy.

Two new special arguments are added: "required_tpu" and "required_gpus".

PiperOrigin-RevId: 197212466
tensorflow/contrib/distribute/python/combinations.py diff | blob | history
Domain: Machine Learning / ML Framework;