提交 586c0e3c authored 作者: Frederic's avatar Frederic

Remove useless case that are not used following code review.

上级 22fe5290
......@@ -1921,11 +1921,7 @@ class MaxAndArgmax(Op):
def infer_shape(self, node, shapes):
ishape, axis_shape = shapes
axis = node.inputs[1]
if axis is None:
return [(), ()]
elif len(axis.data) == 0 and node.inputs[0].ndim:
return [(1,), (1,)]
elif python_all(axis.data == range(node.inputs[0].ndim)):
if python_all(axis.data == range(node.inputs[0].ndim)):
return [(), ()]
rval = tuple([ishape[i] for (i, b) in enumerate(
node.inputs[0].type.broadcastable) if i != axis.data])
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论