提交 8614aac8 编写于 作者: W wjj19950828

deal with comments

上级 9663d20f
...@@ -2583,14 +2583,11 @@ class OpSet9(): ...@@ -2583,14 +2583,11 @@ class OpSet9():
1) == 1 else False 1) == 1 else False
layer_attrs["sorted"] = True if node.get_attr('sorted', layer_attrs["sorted"] = True if node.get_attr('sorted',
1) == 1 else False 1) == 1 else False
is_k_attr = False
k = _const_weight_or_none(val_k) k = _const_weight_or_none(val_k)
if isinstance(k, (list, tuple, np.ndarray)): if isinstance(k, (list, tuple, np.ndarray)):
k = k[0] k = k[0]
if k is not None: if k is not None:
is_k_attr = True
layer_attrs["k"] = k layer_attrs["k"] = k
if is_k_attr:
self.paddle_graph.add_layer( self.paddle_graph.add_layer(
"paddle.topk", "paddle.topk",
inputs={"x": val_x.name}, inputs={"x": val_x.name},
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册