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

Function getGradientForOp

caffe2/python/gradient_checker.py:15–17  ·  view source on GitHub ↗
(op)

Source from the content-addressed store, hash-verified

13
14
15def getGradientForOp(op):
16 return core.GradientRegistry.GetGradientForOp(
17 op, [s + '_grad' for s in op.output])
18
19
20def _get_grad_blob(grad_map, input_to_check):

Callers 1

CheckSimpleMethod · 0.85

Calls 1

GetGradientForOpMethod · 0.80

Tested by

no test coverage detected

Used in the wild real call sites across dependent graphs

searching dependent graphs…