(grads)
| 277 | return all_grads[0] |
| 278 | |
| 279 | def aggregate(grads): |
| 280 | if average: |
| 281 | return tf.multiply(tf.add_n(grads), 1.0 / nr_tower) |
| 282 | else: |
| 283 | return tf.add_n(grads) |
| 284 | |
| 285 | ret = [] |
| 286 | for idx, grad_and_vars in enumerate(zip(*all_grads)): |
no outgoing calls
no test coverage detected