Home
last modified time | relevance | path

Searched defs:update_fn (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/python/training/
Dmoving_averages.py93 def update_fn(v, value): function
181 def _update(strategy, var, update_fn, args): argument
270 def update_fn(v, value, biased_var, local_step): function
/external/tensorflow/tensorflow/python/distribute/
Dvars_test.py89 update_fn = lambda: getattr(v, fn)(update_value) function
135 update_fn = lambda: getattr(v, fn)(update_value) function
186 update_fn = lambda: getattr(v, fn)(update_value) function
650 update_fn = lambda: getattr(v, fn)(update_value) function
698 update_fn = lambda: getattr(v, fn)(update_value) function
766 update_fn = lambda: getattr(v, fn)(update_value) function
810 update_fn = lambda: getattr(v, fn)(update_value) function
Dvalues.py46 def _on_write_update_replica(var, update_fn, value, **kwargs): argument
865 def _update_cross_replica(self, update_fn, value, **kwargs): argument
881 def _update_replica(self, update_fn, value, **kwargs): argument
897 def _update(self, update_fn, value, **kwargs): argument
1036 def _update_replica(self, update_fn, value, **kwargs): argument
1148 def _update_replica(self, update_fn, value, **kwargs): argument
1371 def _update_replica(self, var, update_fn, value, **kwargs): argument
1416 def _update_replica(self, var, update_fn, value, **kwargs): argument
1525 def _update_replica(self, var, update_fn, value, **kwargs): argument
Dmoving_averages_test.py156 def update_fn(v, value): function
Ddistribute_lib_test.py405 def update_fn(): function
Dpacked_distributed_variable.py145 def _update(self, update_fn, value, **kwargs): argument
/external/tflite-support/tensorflow_lite_support/custom_ops/kernel/sentencepiece/
Ddouble_array_trie.h92 callback update_fn) const { in IteratePrefixMatches()
/external/bcc/src/cc/frontends/b/
Dcodegen_llvm.cc626 Function *update_fn = mod_->getFunction("bpf_map_update_elem_"); in emit_table_update() local
656 Function *update_fn = mod_->getFunction("bpf_map_update_elem_"); in emit_table_delete() local
792 Function *update_fn = mod_->getFunction("bpf_map_update_elem_"); in visit_table_index_expr_node() local
/external/tensorflow/tensorflow/python/keras/mixed_precision/
Dautocast_variable.py217 update_fn, argument
242 def _apply_update(self, update_fn, *args, **kwargs): argument
/external/tensorflow/tensorflow/python/ops/
Dstateful_random_ops.py600 def update_fn(v): function