Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
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
772
def
multiple_tpus():
773
devices = context.context().devices()
774
return
len([d
for
d in devices
if
'device:TPU:'
in d]) > 1
775
776
777
class
MultiDeviceTest(xla_test.XLATestCase):
Callers
1
testBasic
Method · 0.85
Calls
2
devices
Method · 0.45
context
Method · 0.45
Tested by
no test coverage detected