Gets all gradients for op.
(grads, op)
| 846 | |
| 847 | |
| 848 | def _GetGrads(grads, op): |
| 849 | """Gets all gradients for op.""" |
| 850 | if op in grads: |
| 851 | return grads[op] |
| 852 | else: |
| 853 | return [[] for _ in xrange(len(op.outputs))] |
| 854 | |
| 855 | |
| 856 | def _AccumulatorShape(inputs): |
no outgoing calls
no test coverage detected