Searched refs:grad_vars (Results 1 – 3 of 3) sorted by relevance
171 grad_vars = [self._grad_map[var] for var in local_vars]172 for g, grad_var in zip(grads, grad_vars):181 for g in grad_vars:186 zip(grad_vars, global_center_vars))198 for grad_var in grad_vars:223 grad_vars = [self._grad_map[var] for var in local_vars]224 if not (local_vars and global_center_vars and grad_vars):229 for g in grad_vars:
502 def pack_range(key, packing, grad_vars, rng): argument516 to_pack = grad_vars[rng[0]:rng[1] + 1]
573 grad_vars = grads[len(inputs):]574 return grad_inputs, grad_vars