Searched refs:assign_add_fn (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/python/distribute/ |
| D | ps_values.py | 136 assign_add_fn = lambda var, *a, **kw: var.assign_add(*a, **kw) function
137 return self._assign_func(f=assign_add_fn, *args, **kwargs)
|
| D | values_util.py | 150 assign_add_fn = lambda var, *a, **kw: var.assign_add(*a, **kw) function
152 update_fn=assign_add_fn,
|
| D | packed_distributed_variable.py | 161 assign_add_fn = lambda var, *a, **kw: var.assign_add(*a, **kw) function
163 update_fn=assign_add_fn,
|
| D | tpu_values.py | 378 assign_add_fn = _make_raw_assign_fn(
381 update_fn=assign_add_fn,
|
| D | input_lib_test.py | 1008 def assign_add_fn(data): function
1012 strategy.run(assign_add_fn, args=(data,))
|
| D | values_test.py | 611 assign_add_fn = lambda var, value: var.assign_add(value) function
612 self.evaluate(distribution.extended.update(v, assign_add_fn, args=(value,)))
|