提交 914ebf7e authored 作者: Tim Cooijmans's avatar Tim Cooijmans

BatchedDot: implement c_code_cache_version

上级 c539a57b
...@@ -2498,7 +2498,8 @@ class BatchedDot(Op): ...@@ -2498,7 +2498,8 @@ class BatchedDot(Op):
""" % locals() """ % locals()
def c_code_cache_version(self): def c_code_cache_version(self):
return None from theano.tensor.blas_headers import blas_header_version
return (1, blas_header_version())
def grad(self, inp, grads): def grad(self, inp, grads):
x, y = inp x, y = inp
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论