batch_size = (
        self.evaluate(serialized).size
        if isinstance(serialized, tensor_lib.Tensor)
        else np.asarray(serialized).size