Searched refs:single_offset (Results 1 – 3 of 3) sorted by relevance
377 offset = partition_info.single_offset(shape)
1780 self.assertEqual(4, partition_info.single_offset([1, 3]))1785 self.assertEqual(0, partition_info.single_offset([9, 3]))
118 def single_offset(self, shape): member in _PartitionInfo