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