Searched refs:single_slice_dim (Results 1 – 2 of 2) sorted by relevance
133 single_slice_dim = self.single_slice_dim(shape)136 if single_slice_dim is None:138 return self.var_offset[single_slice_dim]140 def single_slice_dim(self, shape): member in _PartitionInfo
1794 partition_info.single_slice_dim(None)1798 partition_info.single_slice_dim([1, 2, 3])1802 partition_info.single_slice_dim([6, 3])1806 partition_info.single_slice_dim([1, 1])1810 self.assertEqual(1, partition_info.single_slice_dim([9, 2]))1813 self.assertEqual(0, partition_info.single_slice_dim([2, 3]))