提交 b33e939b authored 作者: Ian Goodfellow's avatar Ian Goodfellow

added connection_pattern for Alloc

上级 09a3c807
......@@ -3106,6 +3106,16 @@ class Alloc(gof.Op):
def infer_shape(self, node, input_shapes):
return [node.inputs[1:]]
def connection_pattern(self, node):
rval = [[True]]
for ipt in node.inputs[1:]:
rval.append([False])
return rval
def grad(self, inputs, grads):
x = inputs[0]
gz = grads[0]
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论