def run_gpu_qr(self, A_val, complete=True): A = theano.tensor.fmatrix("A") fn = theano.function([A], gpu_qr(A, complete=complete), mode=mode_with_gpu) return fn(A_val)