Searched refs:rt_nested_splits_in (Results 1 – 1 of 1) sorted by relevance
35 OpInputList rt_nested_splits_in; in Compute() local37 context, context->input_list("rt_nested_splits", &rt_nested_splits_in)); in Compute()38 const int64 rt_nested_splits_len = rt_nested_splits_in.size(); in Compute()43 rt_nested_splits.push_back(rt_nested_splits_in[i].flat<int64>()); in Compute()135 sparse_dense_shape(0) = rt_nested_splits_in[0].dim_size(0) - 1; in Compute()