Searched refs:rt_nested_splits_len (Results 1 – 1 of 1) sorted by relevance
40 const int rt_nested_splits_len = rt_nested_splits_in.size(); in Compute() local41 DCHECK_GT(rt_nested_splits_len, 0); // Enforced by REGISTER_OP. in Compute()43 rt_nested_splits.reserve(rt_nested_splits_len); in Compute()44 for (int i = 0; i < rt_nested_splits_len; ++i) { in Compute()49 const Tensor& rt_dense_values_in = context->input(rt_nested_splits_len); in Compute()58 std::vector<int64> index_prefix(rt_nested_splits_len); in Compute()66 const int64 indices_len = rt_nested_splits_len + rt_dense_values_in.dims(); in Compute()75 std::vector<int64> pos(rt_nested_splits_len); in Compute()76 int64& final_pos = pos[rt_nested_splits_len - 1]; in Compute()86 for (int dim = rt_nested_splits_len - 2; dim >= 0; --dim) { in Compute()[all …]