Searched refs:rt_nested_splits_in (Results 1 – 1 of 1) sorted by relevance
37 OpInputList rt_nested_splits_in; in Compute() local39 context, context->input_list("rt_nested_splits", &rt_nested_splits_in)); in Compute()40 const int rt_nested_splits_len = rt_nested_splits_in.size(); in Compute()45 rt_nested_splits.push_back(rt_nested_splits_in[i].flat<SPLITS_TYPE>()); in Compute()137 sparse_dense_shape(0) = rt_nested_splits_in[0].dim_size(0) - 1; in Compute()