Started changing Elementwise to prepare to mask out updates during backprop.