-
Notifications
You must be signed in to change notification settings - Fork 3.6k
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.
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
Add UnionShape for SparseTensor TypeProto #3461
Conversation
Signed-off-by: Dmitri Smirnov <dmitrism@microsoft.com>
|
||
namespace ONNX_NAMESPACE { | ||
|
||
void UnionShapeInfo(const TensorShapeProto& source_shape, TensorShapeProto& target_shape) { |
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.
Thanks for the refactoring and moving it into a .cc file. May be useful to add a comment above that source_shape and target_shape are assumed to have same dim_size().
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.
I will remove it from the header
LGTM, thanks! Just a minor comment above. |
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
Signed-off-by: Dmitri Smirnov <dmitrism@microsoft.com>
Signed-off-by: Dmitri Smirnov <dmitrism@microsoft.com>
* Add UnionShape for SparseTensor TypeProto Signed-off-by: Dmitri Smirnov <dmitrism@microsoft.com> * Remove utility function from the header. Signed-off-by: Dmitri Smirnov <dmitrism@microsoft.com> * Document new utility function. Signed-off-by: Dmitri Smirnov <dmitrism@microsoft.com>
* Add UnionShape for SparseTensor TypeProto Signed-off-by: Dmitri Smirnov <dmitrism@microsoft.com> * Remove utility function from the header. Signed-off-by: Dmitri Smirnov <dmitrism@microsoft.com> * Document new utility function. Signed-off-by: Dmitri Smirnov <dmitrism@microsoft.com> Signed-off-by: neginraoof <neginmr@utexas.edu>
This is a missing utility for SparseTensor type proto