def get_partitions(self, x, slices):
        """
        given a partitioning, slices, of an activation buffer, x, determine which axis to slice
        along depending on whether x is a sequential tensor or not
        """
        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]
dataset/ETHPy150Open NervanaSystems/neon/neon/layers/container.py/MergeBroadcast.get_partitions/operator `!=` -> `==`