Searched refs:single_slice_dim (Results 1 – 2 of 2) sorted by relevance
128 single_slice_dim = self.single_slice_dim(shape)131 if single_slice_dim is None:133 return self.var_offset[single_slice_dim]135 def single_slice_dim(self, shape): member in _PartitionInfo
1749 partition_info.single_slice_dim(None)1753 partition_info.single_slice_dim([1, 2, 3])1757 partition_info.single_slice_dim([6, 3])1761 partition_info.single_slice_dim([1, 1])1765 self.assertEqual(1, partition_info.single_slice_dim([9, 2]))1768 self.assertEqual(0, partition_info.single_slice_dim([2, 3]))