Searched defs:assign_fn (Results 1 – 12 of 12) sorted by relevance
| /external/tensorflow/tensorflow/python/distribute/ |
| D | tpu_util.py | 92 def assign_fn(var, value, use_locking=False, name=None, read_value=True): function
|
| D | distribute_lib_test.py | 391 def assign_fn(vv, tt): function
402 def assign_fn(unused_vv, unused_tt): function
|
| D | tpu_replicated_variable.py | 224 assign_fn = lambda var, *a, **ka: var.assign(*a, **ka) function
|
| D | distributed_variable_test.py | 283 assign_fn = lambda var, value: var.assign(value) function
|
| D | values_util.py | 135 assign_fn = lambda var, *a, **kw: var.assign(*a, **kw) function
|
| D | values_test.py | 565 def assign_fn(): function
|
| D | packed_distributed_variable.py | 166 assign_fn = lambda var, *a, **kw: var.assign(*a, **kw) function
|
| D | vars_test.py | 1147 def assign_fn(): function
|
| D | tpu_values.py | 291 assign_fn = lambda v, *a, **ka: v.assign(*a, **ka) function
|
| D | ps_values.py | 159 assign_fn = lambda var, *a, **kw: var.assign(*a, **kw) function
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | variables.py | 3047 def _apply_assign_fn(self, assign_fn, value): argument
3074 assign_fn = lambda var, r_value: var.assign( function
3082 assign_fn = lambda var, r_value: var.assign_add( function
3090 assign_fn = lambda var, r_value: var.assign_sub( function
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | def_function.py | 358 def assign_fn(): function
|