提交 d7cd25c5 authored 作者: Frederic Bastien's avatar Frederic Bastien

Reuse the c code between dtype for gemm, dot22, dot22scalar

上级 8bab8c2c
...@@ -821,6 +821,7 @@ class Gemm(GemmRelated): ...@@ -821,6 +821,7 @@ class Gemm(GemmRelated):
__props__ = ('inplace',) __props__ = ('inplace',)
params_type = ParamsType(inplace=bool_t,) params_type = ParamsType(inplace=bool_t,)
check_input = False
def __init__(self, inplace): def __init__(self, inplace):
self.inplace = inplace self.inplace = inplace
...@@ -1527,6 +1528,7 @@ class Dot22(GemmRelated): ...@@ -1527,6 +1528,7 @@ class Dot22(GemmRelated):
This is a specialization of the more general Dot(). This is a specialization of the more general Dot().
""" """
check_input = False
def make_node(self, x, y): def make_node(self, x, y):
dtypes = ('float16', 'float32', 'float64', 'complex64', 'complex128') dtypes = ('float16', 'float32', 'float64', 'complex64', 'complex128')
...@@ -1789,6 +1791,7 @@ class Dot22Scalar(GemmRelated): ...@@ -1789,6 +1791,7 @@ class Dot22Scalar(GemmRelated):
compute scalar*dot(x,y). compute scalar*dot(x,y).
""" """
check_input = False
def make_node(self, x, y, a): def make_node(self, x, y, a):
if a.ndim != 0: if a.ndim != 0:
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论