MCPcopy Create free account
hub / github.com/alibaba/nann / _resolve_grad_captures

Function _resolve_grad_captures

tensorflow/tensorflow/python/ops/while_v2.py:674–716  ·  view source on GitHub ↗

Returns the tensors to pass as captured inputs to `body_grad_graph`. `body_grad_graph` may have external references to: 1. Its outer graph containing the input gradients. These are left as-is. 2. Accumulators captured from the forward-pass graph. These should have been added as `while_op

(body_graph, body_grad_graph, while_op)

Source from the content-addressed store, hash-verified

source not stored for this graph (policy: none)

Callers 1

_WhileGradFunction · 0.85

Calls 1

appendMethod · 0.45

Tested by

no test coverage detected