Searched refs:assign_fn (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/ |
| D | variables.py | 2624 def _apply_assign_fn(self, assign_fn, value): argument
2645 assign_fn(var, value_list[idx])
2651 assign_fn = lambda var, r_value: var.assign( function
2654 assign_list = self._apply_assign_fn(assign_fn, value)
2660 assign_fn = lambda var, r_value: var.assign_add( function
2663 assign_list = self._apply_assign_fn(assign_fn, value)
2669 assign_fn = lambda var, r_value: var.assign_sub( function
2672 assign_list = self._apply_assign_fn(assign_fn, value)
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | values.py | 707 assign_fn = lambda var, *a, **kw: var.assign(*a, **kw) function
708 return self._assign_func(f=assign_fn, *args, **kwargs)
1032 def assign_fn(var, value, *ar, **kw): function
1044 return self._assign_func(f=assign_fn, *args, **kwargs)
1517 assign_fn = lambda var, *a, **kw: var.assign(*a, **kw) function
1518 return self._assign_func(f=assign_fn, *args, **kwargs)
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | def_function.py | 186 def assign_fn(): function
200 not_assign_fn, assign_fn)
|
| /external/tensorflow/tensorflow/contrib/training/python/training/ |
| D | training_test.py | 427 assign_fn = variables_lib.assign_from_checkpoint_fn(
431 assign_fn(session)
|