MCPcopy Create free account
hub / github.com/pytorch/pytorch / RemapEntry

Class RemapEntry

caffe2/python/core.py:2449–2458  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

2447
2448
2449class RemapEntry:
2450 def __init__(self, blob, device):
2451 self.blob = blob
2452 self.device = device
2453
2454 def __eq__(self, other):
2455 return self.blob == other.blob and self.device == other.device
2456
2457 def __hash__(self):
2458 return hash(self.blob + str(self.device))
2459
2460
2461def InjectCrossDeviceCopies(net, blob_to_device=None, blob_remap=None,

Callers 1

InjectCrossDeviceCopiesFunction · 0.85

Calls

no outgoing calls

Tested by

no test coverage detected

Used in the wild real call sites across dependent graphs

searching dependent graphs…