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()])
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()])
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()])