Searched refs:_flatten_shape (Results 1 – 1 of 1) sorted by relevance
629 def _flatten_shape(s, index): function637 yield from _flatten_shape(sub, subindex)640 return tuple(_flatten_shape(s, index=()))