Searched refs:rt_nested_splits_len (Results 1 – 1 of 1) sorted by relevance
38 const int64 rt_nested_splits_len = rt_nested_splits_in.size(); in Compute() local39 DCHECK_GT(rt_nested_splits_len, 0); // Enforced by REGISTER_OP. in Compute()41 rt_nested_splits.reserve(rt_nested_splits_len); in Compute()42 for (int i = 0; i < rt_nested_splits_len; ++i) { in Compute()47 const Tensor& rt_dense_values_in = context->input(rt_nested_splits_len); in Compute()56 std::vector<int64> index_prefix(rt_nested_splits_len); in Compute()64 const int64 indices_len = rt_nested_splits_len + rt_dense_values_in.dims(); in Compute()73 std::vector<int64> pos(rt_nested_splits_len); in Compute()74 int64& final_pos = pos[rt_nested_splits_len - 1]; in Compute()84 for (int dim = rt_nested_splits_len - 2; dim >= 0; --dim) { in Compute()[all …]