Searched refs:rt_dense_values_in (Results 1 – 1 of 1) sorted by relevance
49 const Tensor& rt_dense_values_in = context->input(rt_nested_splits_len); in Compute() local51 ValidateInputs(rt_nested_splits, rt_dense_values_in)); in Compute()60 MakeIndexSuffixes(rt_dense_values_in.shape()); in Compute()66 const int64 indices_len = rt_nested_splits_len + rt_dense_values_in.dims(); in Compute()121 if (rt_dense_values_in.dims() == 1) { in Compute()122 context->set_output(1, rt_dense_values_in); in Compute()124 Tensor sparse_values_out(rt_dense_values_in.dtype()); in Compute()126 rt_dense_values_in, {rt_dense_values_in.NumElements()}); in Compute()132 int64 ndims = rt_nested_splits_len + rt_dense_values_in.dims(); in Compute()146 for (int dim = 1; dim < rt_dense_values_in.dims(); ++dim) { in Compute()[all …]