OOM when allocating tensor with shape

Note that this is a tensorflow error…