Fix pixel_wise for assign expressions. Add evaluation of if then else expressions.