提交 ea52c340 authored 作者: AndroidCloud's avatar AndroidCloud

updates

上级 6fbae155
...@@ -1157,12 +1157,11 @@ class GpuReshape(HideC, tensor.Reshape): ...@@ -1157,12 +1157,11 @@ class GpuReshape(HideC, tensor.Reshape):
out[0] = x.reshape(tuple(shp)) out[0] = x.reshape(tuple(shp))
def c_code_cache_version(self): def c_code_cache_version(self):
return (1,) return (2,)
def c_code(self, node, name, inputs, outputs, sub): def c_code(self, node, name, inputs, outputs, sub):
x, shape = inputs x, shape = inputs
output, = outputs output, = outputs
new_ndim = self.ndim
sdtype = node.inputs[1].type.dtype_specs()[1] sdtype = node.inputs[1].type.dtype_specs()[1]
fail = sub['fail'] fail = sub['fail']
params = sub['params'] params = sub['params']
......
...@@ -4862,7 +4862,7 @@ class Reshape(Op): ...@@ -4862,7 +4862,7 @@ class Reshape(Op):
for i in xrange(self.ndim)])] for i in xrange(self.ndim)])]
def c_code_cache_version(self): def c_code_cache_version(self):
return (7,) return (8,)
def c_code(self, node, name, inputs, outputs, sub): def c_code(self, node, name, inputs, outputs, sub):
if isinstance(node.inputs[0], TensorVariable): if isinstance(node.inputs[0], TensorVariable):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论