Searched refs:_expand_ellipsis (Results 1 – 1 of 1) sorted by relevance
132 expanded_key_list = _expand_ellipsis(key_list, rt_input.shape.ndims)240 expanded_key_list = _expand_ellipsis(key_list, rt_input.values.shape.ndims)285 def _expand_ellipsis(key_list, num_remaining_dims): function