Searched refs:_row_partitions_for_uniform_shape (Results 1 – 2 of 2) sorted by relevance
258 row_partitions = _row_partitions_for_uniform_shape(1261 return _row_partitions_for_uniform_shape(shape, rank)1275 def _row_partitions_for_uniform_shape(shape, rank): function
819 partitions = structured_tensor._row_partitions_for_uniform_shape(