MCPcopy Create free account
hub / github.com/MegEngine/MegEngine / backward

Method backward

imperative/python/megengine/distributed/functional.py:542–545  ·  view source on GitHub ↗
(self, grad)

Source from the content-addressed store, hash-verified

540 )
541
542 def backward(self, grad):
543 has_grad = _bcast_has_grad(self.group, grad)
544 if has_grad:
545 return scatter(grad, self.group, self.in_device)
546
547
548def gather(

Callers

nothing calls this directly

Calls 2

_bcast_has_gradFunction · 0.85
scatterFunction · 0.70

Tested by

no test coverage detected