Searched defs:leaf_ids (Results 1 – 2 of 2) sorted by relevance
300 auto leaf_ids = output_predictions->tensor<int32, 1>(); in Compute() local335 const Tensor& leaf_ids = context->input(1); in Compute() local359 void UpdateModel(const Tensor& leaf_ids, const TensorInputTarget& target, in UpdateModel()
155 const auto leaf_ids = leaf_ids_tensor.unaligned_flat<int32>(); in UpdateStats() local270 const auto leaf_ids = leaf_ids_tensor.unaligned_flat<int32>(); in Compute() local