Searched refs:testReplicaId (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/python/distribute/ | ||
| D | one_device_strategy_test.py | 49 def testReplicaId(self, distribution): member in OneDeviceStrategyTest |
| D | mirrored_strategy_test.py | 86 def testReplicaId(self, distribution): member in MirroredTwoDeviceDistributionTest |
| /external/tensorflow/tensorflow/compiler/xla/python/ | ||
| D | xla_client_test.py | 664 def testReplicaId(self): member in TestFactory.SingleOpTest |