Split updates for NAG across two theano function calls.