Fix in backward_cpu computation