-
Notifications
You must be signed in to change notification settings - Fork 6.8k
Conversation
Hey @Zha0q1 , Thanks for submitting the PR
CI supported jobs: [windows-gpu, sanity, website, clang, miscellaneous, centos-gpu, edge, windows-cpu, unix-gpu, unix-cpu, centos-cpu] Note: |
@@ -2280,6 +2289,200 @@ def convert_layer_norm(node, **kwargs): | |||
return nodes | |||
|
|||
|
|||
def make_tensor(shape_list, shape_name, initializer, dtype='int64'): |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Please rename to create_tensor() or something, so it's a clear distinction between this and onnx.helper.make_tensor().
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Yeah noted
|
||
if(axis == 0): | ||
nodes += [ | ||
make_node('Reshape', [name+'_slice_0', name+'_void'], [name+'_max_len'], name = '111'), |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Do we need to specify the name parameter?
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Ohhh sorry those were leftovers when I was debugging. Will remove in next push
reverse
)Test cases to be added after onnx test suite is more complete