Ejemplos de Gemv en Python

Ejemplo n.º 1

Archivo: blas.py Proyecto: caglar/Theano

 def make_node(self, y, alpha, A, x, beta):
     Gemv.make_node(self, y, alpha, A, x, beta)
     A = as_gpuarray_variable(A)
     x = as_gpuarray_variable(x)
     y = as_gpuarray_variable(y)
     assert A.dtype == x.dtype == y.dtype
     return Apply(self, [y, alpha, A, x, beta], [y.type()])

Ejemplo n.º 2

Archivo: blas.py Proyecto: MatthieuCourbariaux/Theano

 def make_node(self, y, alpha, A, x, beta):
     res = Gemv.make_node(self, y, alpha, A, x, beta)
     A = as_gpuarray_variable(A)
     x = as_gpuarray_variable(x)
     y = as_gpuarray_variable(y)
     assert A.dtype == x.dtype == y.dtype
     return Apply(self, [y, alpha, A, x, beta], [y.type()])

Ejemplo n.º 3

Archivo: blas.py Proyecto: SamuelZeng/Theano

 def make_node(self, y, alpha, A, x, beta):
     res = Gemv.make_node(self, y, alpha, A, x, beta)
     A = as_gpuarray_variable(A)
     x = as_gpuarray_variable(x)
     y = as_gpuarray_variable(y)
     return Apply(self, [y, alpha, A, x, beta], [y.type()])