提交 d1be796e authored 作者: ricardoV94's avatar ricardoV94 提交者: Ricardo Vieira

Remove useless checks guaranteed by tracks

上级 4c40efa2
......@@ -636,7 +636,6 @@ def local_inplace_ger(fgraph, node):
@node_rewriter([gemm_no_inplace])
def local_gemm_to_gemv(fgraph, node):
"""GEMM acting on row or column matrices -> GEMV."""
if node.op == gemm_no_inplace:
z, a, x, y, b = node.inputs
if z.broadcastable == x.broadcastable == (True, False):
r = gemv_no_inplace(z.dimshuffle(1), a, y.T, x.dimshuffle(1), b)
......@@ -653,7 +652,6 @@ def local_gemm_to_gemv(fgraph, node):
@node_rewriter([gemm_no_inplace])
def local_gemm_to_ger(fgraph, node):
"""GEMM computing an outer-product -> GER."""
if node.op == gemm_no_inplace:
z, a, x, y, b = node.inputs
if x.broadcastable[1] and y.broadcastable[0]:
# x and y are both vectors so this might qualifies for a GER
......@@ -680,12 +678,10 @@ def local_gemm_to_ger(fgraph, node):
return new_out
# TODO: delete this optimization when we have the proper dot->gemm->ger pipeline
# working
# TODO: delete this optimization when we have the proper dot->gemm->ger pipeline working
@node_rewriter([_dot22])
def local_dot22_to_ger_or_gemv(fgraph, node):
"""dot22 computing an outer-product -> GER."""
if node.op == _dot22:
x, y = node.inputs
xb = x.broadcastable
yb = y.broadcastable
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论