Searched refs:rt_dense_values_in (Results 1 – 1 of 1) sorted by relevance
47 const Tensor& rt_dense_values_in = context->input(rt_nested_splits_len); in Compute() local49 ValidateInputs(rt_nested_splits, rt_dense_values_in)); in Compute()58 MakeIndexSuffixes(rt_dense_values_in.shape()); in Compute()64 const int64 indices_len = rt_nested_splits_len + rt_dense_values_in.dims(); in Compute()119 if (rt_dense_values_in.dims() == 1) { in Compute()120 context->set_output(1, rt_dense_values_in); in Compute()122 Tensor sparse_values_out(rt_dense_values_in.dtype()); in Compute()124 rt_dense_values_in, {rt_dense_values_in.NumElements()}); in Compute()130 int64 ndims = rt_nested_splits_len + rt_dense_values_in.dims(); in Compute()144 for (int dim = 1; dim < rt_dense_values_in.dims(); ++dim) { in Compute()[all …]