def get_partitions(self, x, slices):
        if x.shape[-1] != self.be.bsz:  # This is the sequential case
            return [x[:, sl] for sl in slices]
        else:
            return [x[sl] for sl in slices]
given a partitioning, slices, of an activation buffer, x, determine which axis to slice
dataset/ETHPy150Open NervanaSystems/neon/neon/layers/container.py/MergeBroadcast.get_partitions