Searched refs:assign_sub_fn (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/python/distribute/ |
| D | ps_values.py | 132 assign_sub_fn = lambda var, *a, **kw: var.assign_sub(*a, **kw) function
133 return self._assign_func(f=assign_sub_fn, *args, **kwargs)
|
| D | values_util.py | 161 assign_sub_fn = lambda var, *a, **kw: var.assign_sub(*a, **kw) function
163 update_fn=assign_sub_fn,
|
| D | packed_distributed_variable.py | 152 assign_sub_fn = lambda var, *a, **kw: var.assign_sub(*a, **kw) function
154 update_fn=assign_sub_fn,
|
| D | tpu_values.py | 367 assign_sub_fn = _make_raw_assign_fn(
370 update_fn=assign_sub_fn,
|
| D | values_test.py | 615 assign_sub_fn = lambda var, value: var.assign_sub(value) function
616 self.evaluate(distribution.extended.update(v, assign_sub_fn, args=(value,)))
|