-
Notifications
You must be signed in to change notification settings - Fork 6.7k
[v1.x] [Backport] Add checks in batchnorm's infer shape #20444
Conversation
|
Hey @bgawrych , Thanks for submitting the PR
CI supported jobs: [windows-cpu, unix-gpu, clang, unix-cpu, miscellaneous, centos-cpu, edge, sanity, centos-gpu, windows-gpu, website] Note: |
0f7b51b to
3ce2ef2
Compare
|
@mxnet-bot run ci [all] |
|
Jenkins CI successfully triggered : [website, clang, windows-gpu, unix-gpu, sanity, unix-cpu, windows-cpu, centos-gpu, miscellaneous, centos-cpu, edge] |
|
@mxnet-bot run ci [centos-gpu, unix-gpu] |
|
Jenkins CI successfully triggered : [centos-gpu, unix-gpu] |
bartekkuncer
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.
Please apply formatter to the changed code.
Done for my changes - for whole files (src/operator/nn/batch_norm.cc and tests/cpp/operator/batchnorm_test.cc) probably would be better to make it in other commit with more formatting changes |
|
@mxnet-bot run ci [unix-gpu] |
|
Jenkins CI successfully triggered : [unix-gpu] |
bartekkuncer
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
Add checks to infer shape to avoid passing invalid tensors. Connected to issue: #19065
Backport of: #20415