MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / multiple_tpus

Function multiple_tpus

tensorflow/compiler/tests/eager_test.py:772–774  ·  view source on GitHub ↗
()

Source from the content-addressed store, hash-verified

770
771
772def multiple_tpus():
773 devices = context.context().devices()
774 return len([d for d in devices if 'device:TPU:' in d]) > 1
775
776
777class MultiDeviceTest(xla_test.XLATestCase):

Callers 1

testBasicMethod · 0.85

Calls 2

devicesMethod · 0.45
contextMethod · 0.45

Tested by

no test coverage detected