提交 15c64971 authored 作者: Brandon T. Willard's avatar Brandon T. Willard 提交者: Brandon T. Willard

Implement DeepCopyOp.infer_shape

上级 477f4e5d
......@@ -206,6 +206,9 @@ class DeepCopyOp(COp):
# Else, no C code
raise NotImplementedError()
def infer_shape(self, fgraph, node, input_shapes):
return input_shapes
deep_copy_op = DeepCopyOp()
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论