Searched refs:leaf_id (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/tensor_forest/ |
| D | prediction_ops.cc | 68 const int32 leaf_id = in Compute() local
70 set_output_value(example_id, leaf_id, decision_tree_resource, &out); in Compute()
77 void set_output_value(const int32 example_id, const int32 leaf_id, in set_output_value() argument
81 const float logit = decision_tree_resource->get_prediction(leaf_id, j); in set_output_value()
|
| /external/tensorflow/tensorflow/contrib/tensor_forest/kernels/ |
| D | stats_ops.cc | 163 int32 leaf_id; in UpdateStats() local
167 std::tie(leaf_id, example_id) = waiting.front(); in UpdateStats()
171 leaf_id = leaf_ids(i); in UpdateStats()
175 const std::unique_ptr<mutex>& leaf_lock = (*locks)[leaf_id]; in UpdateStats()
180 waiting.emplace(leaf_id, example_id); in UpdateStats()
187 data, &target, {example_id}, leaf_id, &is_finished); in UpdateStats()
191 ready_to_split->insert(leaf_id); in UpdateStats()
211 int32 leaf_id = it->first; in UpdateStatsCollated() local
214 data, &target, it->second, leaf_id, &is_finished); in UpdateStatsCollated()
217 ready_to_split->insert(leaf_id); in UpdateStatsCollated()
|
| /external/tensorflow/tensorflow/python/module/ |
| D | module.py | 309 leaf_id = id(leaf)
310 if leaf_id in seen:
312 seen.add(leaf_id)
|
| /external/bcc/src/cc/frontends/b/ |
| D | type_check.cc | 261 …auto result = make_unique<StructVariableDeclStmtNode>(table->leaf_id()->copy(), make_unique<IdentE… in check_lookup_method()
354 …return mkstatus_(n, "Field %s is not a member of %s", n->sub_->c_str(), n->table_->leaf_id()->c_st… in visit_table_index_expr_node()
516 n->leaf_type_ = scopes_->top_struct()->lookup(n->leaf_id()->name_, true); in visit_table_decl_stmt_node()
518 return mkstatus_(n, "Table leaf type %s undefined", n->leaf_id()->c_str()); in visit_table_decl_stmt_node()
519 n->leaf_id()->bit_width_ = n->leaf_type_->bit_width_; in visit_table_decl_stmt_node()
|
| D | codegen_llvm.cc | 601 if (table->leaf_id()->name_ != arg1_type->struct_id_->name_) { in emit_table_lookup()
602 return mkstatus_(n, "lookup pointer type mismatch %s != %s", table->leaf_id()->c_str(), in emit_table_lookup()
824 B.CreateMemSet(leaf_ptr, B.getInt8(0), B.getInt64(n->table_->leaf_id()->bit_width_ >> 3), 1); in visit_table_index_expr_node()
1087 auto leaf = scopes_->top_struct()->lookup(n->leaf_id()->name_, /*search_local*/true); in visit_table_decl_stmt_node()
1088 if (!leaf) return mkstatus_(n, "cannot find leaf %s", n->leaf_id()->name_.c_str()); in visit_table_decl_stmt_node()
|
| D | node.h | 593 IdentExprNode * leaf_id() { return templates_.at(1).get(); } in leaf_id() function
|