I know you copy/pasted the CudaNdarray optimization on this, but why the 2 returns? I think it should be continue. Otherwise, we won't try to move all the Theano function inputs to the GPU.