Searched refs:_row_partitions_for_ragged_tensor (Results 1 – 2 of 2) sorted by relevance
1242 value_row_partitions = _row_partitions_for_ragged_tensor(value, rank, dtype)1264 def _row_partitions_for_ragged_tensor(value, rank, dtype): function
887 partitions = structured_tensor._row_partitions_for_ragged_tensor(