Searched refs:_nested_row_partitions (Results 1 – 2 of 2) sorted by relevance
919 def _nested_row_partitions(self): member in RaggedTensor1470 for i, partition in enumerate(self._nested_row_partitions):2818 partitions = rt_input._nested_row_partitions # pylint: disable=protected-access
1267 value_row_partitions = value._nested_row_partitions[:rank - 1] # pylint: disable=protected-access