Home
last modified time | relevance | path

Searched refs:assign_fn (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/python/distribute/
Dtpu_values.py63 …def assign_fn(var, value, use_locking=False, name=None, read_value=True): # pylint: disable=missi… function
73 return assign_fn
389 assign_fn = _make_raw_assign_fn(gen_resource_variable_ops.assign_variable_op)
391 update_fn=assign_fn,
Ddistribute_lib_test.py374 def assign_fn(vv, tt): function
377 dist.extended.update(v, assign_fn, (t,))
385 def assign_fn(unused_vv, unused_tt): function
390 dist.extended.update(v, assign_fn, (t,))
Dps_values.py140 assign_fn = lambda var, *a, **kw: var.assign(*a, **kw) function
141 return self._assign_func(f=assign_fn, *args, **kwargs)
Dvalues_util.py139 assign_fn = lambda var, *a, **kw: var.assign(*a, **kw) function
141 update_fn=assign_fn,
Dpacked_distributed_variable.py170 assign_fn = lambda var, *a, **kw: var.assign(*a, **kw) function
172 update_fn=assign_fn,
Dvars_test.py1090 def assign_fn(): function
1100 strategy.run(assign_fn)
Dvalues_test.py607 assign_fn = lambda var, value: var.assign(value) function
608 self.evaluate(distribution.extended.update(v, assign_fn, args=(value,)))
/external/tensorflow/tensorflow/python/ops/
Dvariables.py3053 def _apply_assign_fn(self, assign_fn, value): argument
3074 assign_fn(var, value_list[idx])
3080 assign_fn = lambda var, r_value: var.assign( function
3082 assign_list = self._apply_assign_fn(assign_fn, value)
3088 assign_fn = lambda var, r_value: var.assign_add( function
3090 assign_list = self._apply_assign_fn(assign_fn, value)
3096 assign_fn = lambda var, r_value: var.assign_sub( function
3098 assign_list = self._apply_assign_fn(assign_fn, value)
/external/tensorflow/tensorflow/python/eager/
Ddef_function.py311 def assign_fn(): function
331 not_assign_fn, assign_fn)
/external/rust/crates/ppv-lite86/src/x86_64/
Dsse2.rs1552 ($vec:ident, $Assign:ident, $assign_fn:ident, $bin_fn:ident) => {
1558 fn $assign_fn(&mut self, rhs: Self) {