Searched refs:dimension_ids_list (Results 1 – 1 of 1) sorted by relevance
254 OpInputList dimension_ids_list; in Compute() local263 context->input_list("dimension_ids", &dimension_ids_list)); in Compute()290 thresholds_list, dimension_ids_list, in Compute()392 const OpInputList& thresholds_list, const OpInputList& dimension_ids_list, in FindBestSplitsPerNode() argument403 const auto& dimension_ids = dimension_ids_list[group_idx].vec<int32>(); in FindBestSplitsPerNode()