MCPcopy Create free account
hub / github.com/Anoise/WTFlib / cuda

Method cuda

LDPS_Graph/layers/utils.py:288–290  ·  view source on GitHub ↗
(self)

Source from the content-addressed store, hash-verified

286 return x
287
288 def cuda(self):
289 self.mean = self.mean.cuda()
290 self.std = self.std.cuda()
291
292 def cpu(self):
293 self.mean = self.mean.cpu()

Callers

nothing calls this directly

Calls 1

cudaMethod · 0.45

Tested by

no test coverage detected