提交 bce7c0a2 authored 作者: Samira Shabanian's avatar Samira Shabanian

added tensor.blas.Dot22

上级 ba15e178
...@@ -6189,7 +6189,8 @@ def test_local_sumsqr2dot(): ...@@ -6189,7 +6189,8 @@ def test_local_sumsqr2dot():
f_test = function([W,G], T.dot(T.sqr(G), T.sqr(W).sum(axis=0)), mode=MODE) f_test = function([W,G], T.dot(T.sqr(G), T.sqr(W).sum(axis=0)), mode=MODE)
assert numpy.allclose(f_val, f_test(w_val, g_val)) assert numpy.allclose(f_val, f_test(w_val, g_val))
assert any(isinstance(n.op, (tensor.basic.Dot, tensor.blas.Gemv, tensor.blas_c.CGemv)) assert any(isinstance(n.op, (tensor.basic.Dot, tensor.blas.Dot22,
tensor.blas.Gemv, tensor.blas_c.CGemv))
for n in f.maker.fgraph.toposort()) for n in f.maker.fgraph.toposort())
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论