Searched refs:leaf_ids (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/contrib/tensor_forest/kernels/ |
| D | model_ops.cc | 300 auto leaf_ids = output_predictions->tensor<int32, 1>(); in Compute() local
302 auto set_leaf_ids = [&leaf_ids](int32 i, int32 id) { leaf_ids(i) = id; }; in Compute()
335 const Tensor& leaf_ids = context->input(1); in Compute() local
356 UpdateModel(leaf_ids, target, 0, num_data, decision_tree_resource); in Compute()
359 void UpdateModel(const Tensor& leaf_ids, const TensorInputTarget& target, in UpdateModel() argument
362 const auto leaves = leaf_ids.unaligned_flat<int32>(); in UpdateModel()
|
| D | stats_ops.cc | 155 const auto leaf_ids = leaf_ids_tensor.unaligned_flat<int32>(); in UpdateStats() local
171 leaf_id = leaf_ids(i); in UpdateStats()
270 const auto leaf_ids = leaf_ids_tensor.unaligned_flat<int32>(); in Compute() local
280 leaf_examples[leaf_ids(i)].push_back(i); in Compute()
284 const int32 id = leaf_ids(i); in Compute()
|
| /external/tensorflow/tensorflow/contrib/tensor_forest/python/ |
| D | tensor_forest.py | 624 leaf_ids = model_ops.traverse_tree_v4(
635 leaf_ids,
649 leaf_ids,
|