Home
last modified time | relevance | path

Searched defs:assign_fn (Results 1 – 12 of 12) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/python/distribute/
H A Ddistribute_lib_test.py391 def assign_fn(vv, tt): function
402 def assign_fn(unused_vv, unused_tt): function
H A Dtpu_util.py92 def assign_fn(var, value, use_locking=False, name=None, read_value=True): function
H A Ddistributed_variable_test.py283 assign_fn = lambda var, value: var.assign(value) function
H A Dtpu_replicated_variable.py224 assign_fn = lambda var, *a, **ka: var.assign(*a, **ka) function
H A Dvalues_util.py135 assign_fn = lambda var, *a, **kw: var.assign(*a, **kw) function
H A Dvalues_test.py565 def assign_fn(): function
H A Dpacked_distributed_variable.py166 assign_fn = lambda var, *a, **kw: var.assign(*a, **kw) function
H A Dvars_test.py1147 def assign_fn(): function
H A Dtpu_values.py291 assign_fn = lambda v, *a, **ka: v.assign(*a, **ka) function
H A Dps_values.py159 assign_fn = lambda var, *a, **kw: var.assign(*a, **kw) function
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/
H A Dvariables.py3047 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
/aosp_15_r20/external/tensorflow/tensorflow/python/eager/
H A Ddef_function.py358 def assign_fn(): function