Example #1
0
 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)
Example #2
0
 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)