MCPcopy
hub / github.com/THUDM/CogDL / cuda

Method cuda

cogdl/data/data.py:130–131  ·  view source on GitHub ↗
(self, *keys)

Source from the content-addressed store, hash-verified

128 return self.apply(lambda x: x.to(device), *keys)
129
130 def cuda(self, *keys):
131 return self.apply(lambda x: x.cuda(), *keys)
132
133
134class Adjacency(BaseGraph):

Callers 1

forwardMethod · 0.80

Calls 1

applyMethod · 0.95

Tested by

no test coverage detected