| /external/tensorflow/tensorflow/contrib/mixed_precision/python/ |
| D | loss_scale_manager_test.py | 43 update_fn = lambda: lsm.update_loss_scale(itr.get_next()) function
47 update_op = update_fn()
50 update_fn()
72 update_fn = lambda: lsm.update_loss_scale(itr.get_next()) function
78 update_op = update_fn()
81 update_fn()
|
| /external/libtextclassifier/utils/sentencepiece/ |
| D | sorted_strings_table.cc | 25 StringPiece input, const std::function<void(TrieMatch)>& update_fn) const { in GatherPrefixMatches()
68 update_fn(TrieMatch(/*id=*/left, in GatherPrefixMatches()
89 update_fn(TrieMatch(/*id=*/i, in GatherPrefixMatches()
|
| D | double_array_trie.cc | 23 StringPiece input, const std::function<void(TrieMatch)>& update_fn) const { in GatherPrefixMatches()
49 update_fn(TrieMatch(/*id=*/value(pos), /*match_length=*/i + 1)); in GatherPrefixMatches()
|
| D | sorted_strings_table.h | 57 StringPiece input, const std::function<void(TrieMatch)>& update_fn) const;
|
| D | double_array_trie.h | 77 StringPiece input, const std::function<void(TrieMatch)>& update_fn) const;
|
| /external/tensorflow/tensorflow/contrib/opt/python/training/ |
| D | weight_decay_optimizers_test.py | 58 def doTest(self, optimizer, update_fn, optimizer_name, slot_name, argument
110 var0_np, m0, v0 = update_fn(var0_np, grads0_np, t=t, m=m0, v=v0)
111 var1_np, m1, v1 = update_fn(var1_np, grads1_np, t=t, m=m1, v=v1)
|
| /external/tensorflow/tensorflow/python/training/ |
| D | moving_averages.py | 82 def update_fn(v, value, decay=decay): function
101 return strategy.extended.update(v, update_fn, args=(value,))
106 return strategy.extended.update(variable, update_fn, args=(value,))
|
| /external/tensorflow/tensorflow/python/keras/utils/ |
| D | metrics_utils.py | 347 update_fn = ConfusionMatrix.FALSE_NEGATIVES in variables_to_update
349 if update_fn or update_tn:
|
| /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/contrib/factorization/python/ops/ |
| D | wals.py | 230 def create_axis_ops(sp_input, num_items, update_fn, axis_name): argument
258 _, update_op, loss, reg, sum_weights = update_fn(sp_input)
|