Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Fixed nonzero op #808

Merged
merged 14 commits into from Jun 8, 2022
Merged

Fixed nonzero op #808

Changes from all commits
Commits
Show all changes
14 commits
Select commit Hold shift + click to select a range
e754c85
Support bigbird model
wjj19950828 May 7, 2022
094329b
resolve conflict
wjj19950828 May 10, 2022
dc0099f
Merge remote-tracking branch 'upstream/develop' into Support_HF_ops
wjj19950828 May 10, 2022
d5c2a8b
Merge remote-tracking branch 'upstream/develop' into Support_GPT2
wjj19950828 May 12, 2022
10969ba
Support GPT2
wjj19950828 May 12, 2022
4e29cc7
rm useless code
wjj19950828 May 12, 2022
0afa879
deal with comments
wjj19950828 May 26, 2022
907c1f5
Merge remote-tracking branch 'upstream/develop' into Support_GPT2
wjj19950828 May 26, 2022
0bd1269
Merge remote-tracking branch 'upstream/develop' into fixed_topk
wjj19950828 Jun 1, 2022
9663d20
fixed topk
wjj19950828 Jun 1, 2022
8614aac
deal with comments
wjj19950828 Jun 6, 2022
a437d05
add notes
wjj19950828 Jun 6, 2022
005a420
Merge remote-tracking branch 'upstream/develop' into fixed_topk
wjj19950828 Jun 7, 2022
06a3deb
fixed nonzero
wjj19950828 Jun 7, 2022
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Jump to
Jump to file
Failed to load files.
Diff view
Diff view
31 changes: 7 additions & 24 deletions x2paddle/op_mapper/onnx2paddle/opset9/opset.py
View file
Open in desktop
Expand Up @@ -1880,30 +1880,13 @@ def Where(self, node):
@print_mapping_info
def NonZero(self, node):
val_x = self.graph.get_input_node(node, idx=0, copy=True)
val_x_dim = len(val_x.out_shapes[0])
if val_x_dim == 1:
self.paddle_graph.add_layer(
"paddle.nonzero",
inputs={"x": val_x.name},
outputs=[val_x.name])
self.paddle_graph.add_layer(
"paddle.transpose",
inputs={"x": val_x.name},
outputs=[node.layer_name],
perm=[1, 0])
if val_x_dim > 1:
self.paddle_graph.add_layer(
"paddle.nonzero",
inputs={"x": val_x.name},
outputs=[val_x.name])
self.paddle_graph.add_layer(
"paddle.split",
inputs={"x": val_x.name},
outputs=[val_x.name],
num_or_sections=1,
axis=val_x_dim)
self.paddle_graph.add_layer(
"paddle.concat", inputs={"x": val_x.name}, outputs=[node.name])
self.paddle_graph.add_layer(
"paddle.nonzero",
inputs={"x": val_x.name},
outputs=[val_x.name],
as_tuple=True)
self.paddle_graph.add_layer(
"paddle.concat", inputs={"x": val_x.name}, outputs=[node.name])

@print_mapping_info
def Identity(self, node):
Expand Down