Sets gradient "grad" in "grads" for tensor "t".
(grads, t, grad)
| 807 | |
| 808 | |
| 809 | def _SetGrad(grads, t, grad): |
| 810 | """Sets gradient "grad" in "grads" for tensor "t".""" |
| 811 | op = t.op |
| 812 | op_grads = grads.get(op) |
| 813 | if not op_grads: |
| 814 | op_grads = [[] for _ in xrange(len(op.outputs))] |
| 815 | grads[op] = op_grads |
| 816 | t_grads = op_grads[t.value_index] |
| 817 | if isinstance(t_grads, list): |
| 818 | t_grads.append(grad) |
| 819 | else: |
| 820 | assert control_flow_util.IsLoopSwitch(op) |
| 821 | op_grads[t.value_index] = grad |
| 822 | |
| 823 | |
| 824 | def _GetGrad(grads, t, unconnected_gradients): |
no test coverage detected