提交 f0ddb0ea authored 作者: Frederic's avatar Frederic

Fix c code versioning of Conv3d op.

The Attribute Error is catched in cmodule_key
上级 d86ac0d3
...@@ -2,7 +2,7 @@ import theano ...@@ -2,7 +2,7 @@ import theano
from theano.tensor import basic as T from theano.tensor import basic as T
import numpy as N import numpy as N
#from util import strutil #from util import strutil
from theano.tensor.blas_headers import blas_header_text from theano.tensor.blas_headers import blas_header_text, blas_header_version
from theano.tensor.blas import ldflags from theano.tensor.blas import ldflags
from theano.misc import strutil from theano.misc import strutil
from theano.gradient import grad_undefined from theano.gradient import grad_undefined
...@@ -51,8 +51,7 @@ class Conv3D(theano.Op): ...@@ -51,8 +51,7 @@ class Conv3D(theano.Op):
return "Conv3D" return "Conv3D"
def c_code_cache_version(self): def c_code_cache_version(self):
return (3, blas_header_text.version) return (3, blas_header_version())
def make_node(self, V, W, b, d): def make_node(self, V, W, b, d):
""" """
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论