set_subtensor.

This was the cause of the 3 failing test in scan. I also revised the entire
optimization, writing a bit more comments/