| /external/tensorflow/tensorflow/python/distribute/ |
| D | packed_distributed_variable.py | 145 def _update(self, update_fn, value, **kwargs): argument
147 return update_fn(self.get_var_on_current_device(), value, **kwargs)
149 return update_fn(super(PackedDistributedVariable, self), value, **kwargs)
154 update_fn=assign_sub_fn,
163 update_fn=assign_add_fn,
172 update_fn=assign_fn,
181 update_fn=scatter_sub_fn,
189 update_fn=scatter_add_fn,
197 update_fn=scatter_mul_fn,
205 update_fn=scatter_div_fn,
[all …]
|
| D | vars_test.py | 89 update_fn = lambda: getattr(v, fn)(update_value) function
91 return update_fn()
94 update_fn = def_function.function(update_fn)
96 distribution.run(update_fn))
135 update_fn = lambda: getattr(v, fn)(update_value) function
137 return update_fn()
140 update_fn = def_function.function(update_fn)
142 distribution.run(update_fn))
186 update_fn = lambda: getattr(v, fn)(update_value) function
188 return update_fn()
[all …]
|
| D | values_util.py | 141 update_fn=assign_fn,
152 update_fn=assign_add_fn,
163 update_fn=assign_sub_fn,
228 update_fn=scatter_sub_fn,
237 update_fn=scatter_add_fn,
246 update_fn=scatter_mul_fn,
255 update_fn=scatter_div_fn,
264 update_fn=scatter_min_fn,
273 update_fn=scatter_max_fn,
282 update_fn=scatter_update_fn,
|
| D | values.py | 46 def _on_write_update_replica(var, update_fn, value, **kwargs): argument
49 …return update_fn(var._get_on_device_or_primary(), value, **kwargs) # pylint: disable=protected-ac…
70 return var._update_cross_replica(update_fn, v, **kwargs) # pylint: disable=protected-access
865 def _update_cross_replica(self, update_fn, value, **kwargs): argument
879 self, update_fn, args=(value,), kwargs=kwargs, group=True)
881 def _update_replica(self, update_fn, value, **kwargs): argument
894 …return self._policy._update_replica(self, update_fn, value, **kwargs) # pylint: disable=protected…
897 def _update(self, update_fn, value, **kwargs): argument
918 return update_fn(self._primary, value, **kwargs)
924 return update_fn(replica_value, value, **kwargs)
[all …]
|
| D | moving_averages_test.py | 156 def update_fn(v, value): function
162 distribution.extended.update(var1, update_fn, ([1.0, 1.0],))
|
| D | tpu_values.py | 370 update_fn=assign_sub_fn,
381 update_fn=assign_add_fn,
391 update_fn=assign_fn,
|
| D | distribute_lib_test.py | 405 def update_fn(): function
410 dist.extended.update_non_slot(t, update_fn)
|
| /external/tensorflow/tensorflow/python/training/ |
| D | moving_averages.py | 93 def update_fn(v, value): function
100 return _update(strategy, v, update_fn, args=(value,))
181 def _update(strategy, var, update_fn, args): argument
190 return update_fn(var, *args)
192 return strategy.extended.update(var, update_fn, args)
270 def update_fn(v, value, biased_var, local_step): function
281 strategy, unbiased_var, update_fn, args=(value, biased_var, local_step))
|
| /external/libtextclassifier/native/utils/container/ |
| D | sorted-strings-table.cc | 27 StringPiece input, const std::function<void(Match)>& update_fn) const { in GatherPrefixMatches()
72 update_fn(Match(/*id=*/left, in GatherPrefixMatches()
94 update_fn(Match(/*id=*/i, in GatherPrefixMatches()
|
| D | double-array-trie.cc | 24 StringPiece input, const std::function<void(Match)>& update_fn) const { in GatherPrefixMatches()
50 update_fn(Match(/*id=*/value(pos), /*match_length=*/i + 1)); in GatherPrefixMatches()
|
| D | sorted-strings-table.h | 57 const std::function<void(Match)>& update_fn) const;
|
| D | double-array-trie.h | 77 const std::function<void(Match)>& update_fn) const;
|
| /external/tflite-support/tensorflow_lite_support/custom_ops/kernel/sentencepiece/ |
| D | double_array_trie.h | 59 callback update_fn) const;
92 callback update_fn) const { in IteratePrefixMatches() argument
110 update_fn(Match(value(pos), i + 1)); in IteratePrefixMatches()
|
| /external/tensorflow/tensorflow/python/keras/mixed_precision/ |
| D | autocast_variable.py | 217 update_fn, argument
226 assign_op = update_fn(value, use_locking, name, False)
237 assign_var = update_fn(value, use_locking, name, read_value)
242 def _apply_update(self, update_fn, *args, **kwargs): argument
243 update_var = update_fn(*args, **kwargs)
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | stateful_random_ops.py | 600 def update_fn(v): function
607 return update_fn(self.state)
616 self.state, update_fn)
617 return update_fn(self.state)
|
| /external/bcc/src/cc/frontends/b/ |
| D | codegen_llvm.cc | 626 Function *update_fn = mod_->getFunction("bpf_map_update_elem_"); in emit_table_update() local
627 if (!update_fn) return mkstatus_(n, "bpf_map_update_elem_ undefined"); in emit_table_update()
640 …expr_ = B.CreateCall(update_fn, vector<Value *>({pseudo_map_fd, key_ptr, value_ptr, B.getInt64(BPF… in emit_table_update()
656 Function *update_fn = mod_->getFunction("bpf_map_update_elem_"); in emit_table_delete() local
657 if (!update_fn) return mkstatus_(n, "bpf_map_update_elem_ undefined"); in emit_table_delete()
667 expr_ = B.CreateCall(update_fn, vector<Value *>({pseudo_map_fd, key_ptr})); in emit_table_delete()
792 Function *update_fn = mod_->getFunction("bpf_map_update_elem_"); in visit_table_index_expr_node() local
793 if (!update_fn) return mkstatus_(n, "bpf_map_update_elem_ undefined"); in visit_table_index_expr_node()
826 …B.CreateCall(update_fn, vector<Value *>({pseudo_map_fd, key_ptr, leaf_ptr, B.getInt64(BPF_NOEXIST)… in visit_table_index_expr_node()
|
| /external/tensorflow/tensorflow/python/keras/utils/ |
| D | metrics_utils.py | 438 update_fn = ConfusionMatrix.FALSE_NEGATIVES in variables_to_update
440 if update_fn or update_tn:
|