提交 abd4aff1 authored 作者: Razvan Pascanu's avatar Razvan Pascanu

Merge pull request #3 from lamblin/c_viewop

Fixes after code review
......@@ -70,9 +70,8 @@ class ViewOp(gof.Op):
"when calling register_deep_copy_op_c_code." % t,
stacklevel=2)
return ()
version.append((str(t), v))
version.append((str(t), v))
print 'view version:', version
return tuple(version)
def infer_shape(self, node, input_shapes):
......@@ -161,9 +160,8 @@ class DeepCopyOp(gof.Op):
"when calling register_OutputGuard_c_code." % t,
stacklevel=2)
return ()
version.append((str(t), v))
version.append((str(t), v))
print 'deeop copy version:', version
return tuple(version)
def c_code(self, node, name, inames, onames, sub):
......
......@@ -146,7 +146,7 @@ class SparseType(gof.Type):
def is_valid_value(self, a):
return scipy.sparse.issparse(a) and (a.format == self.format)
# Register CudaNdarrayType's C code for ViewOp.
# Register SparseType's C code for ViewOp.
theano.compile.register_view_op_c_code(
SparseType,
"""
......
......@@ -56,7 +56,7 @@ class RandomStateType(gof.Type):
return False
return True
# Register CudaNdarrayType's C code for ViewOp.
# Register RandomStateType's C code for ViewOp.
theano.compile.register_view_op_c_code(
RandomStateType,
"""
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论