Searched refs:grad_and_vars (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/python/distribute/ |
| D | cross_device_utils.py | 134 def aggregate_single_gradient_using_copy(grad_and_vars, use_mean, argument
154 grads = [g for g, _ in grad_and_vars]
160 v = grad_and_vars[0][1]
379 def sum_grad_and_var_all_reduce(grad_and_vars, argument
389 scaled_grads = [g for g, _ in grad_and_vars]
415 for (_, v), g in zip(grad_and_vars, summed_grads):
452 for grad_and_vars in zip(*replica_grads):
455 grad_and_vars, num_workers, alg, gpu_indices, aux_devices
|
| D | cross_device_ops.py | 1041 for grad_and_vars in chunk:
1042 scaled_grads = [g for g, _ in grad_and_vars]
1047 for (_, v), g in zip(grad_and_vars, collective_reduced):
|
| /external/tensorflow/tensorflow/contrib/opt/python/training/ |
| D | drop_stale_gradient_optimizer_test.py | 84 grad_and_vars = stale_check_opt.compute_gradients(loss)
86 grad_and_vars = stale_check_opt.compute_gradients(loss)
87 with ops.control_dependencies([t[0] for t in grad_and_vars]):
93 [stale_check_opt.apply_gradients(grad_and_vars, global_step)]):
100 grad_and_vars, global_step)]):
|