-
Notifications
You must be signed in to change notification settings - Fork 6.7k
[v1.x] Onnx export support for batch_dot #19775
Conversation
|
Hey @waytrue17 , Thanks for submitting the PR
CI supported jobs: [windows-gpu, website, centos-cpu, edge, unix-gpu, sanity, windows-cpu, unix-cpu, miscellaneous, centos-gpu, clang] Note: |
josephevans
left a comment
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.
Just a few small issues.
|
|
||
| if transpose_a == 'False': | ||
| nodes += [ | ||
| make_node('Add', [lhs, name+'_0f'], [name+'_lhs']), |
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.
can we get rid of this add by writing four make_node('MatMul', [name+'_lhs', name+'_rhs'], [name]) in each case? this drags performance down
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.
Addressed
|
@mxnet-bot run ci [unix-gpu] |
|
Jenkins CI successfully triggered : [unix-gpu] |
Zha0q1
left a comment
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.
LGTM!
Description
Onnx export support for batch_dot
Checklist
Essentials
Changes
Comments