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
Correct Tensor order for dilation2D #30277
Correct Tensor order for dilation2D #30277
Conversation
`gen_nn_ops.dilation2d` seems to be in `NHWC` while the parent function was asking for `NCHW`. I corrected the doc and the check.
Thanks for your pull request. It looks like this may be your first contribution to a Google open source project (if not, look below for help). Before we can look at your pull request, you'll need to sign a Contributor License Agreement (CLA). 📝 Please visit https://cla.developers.google.com/ to sign. Once you've signed (or fixed any issues), please reply here (e.g. What to do if you already signed the CLAIndividual signers
Corporate signers
ℹ️ Googlers: Go here for more info. |
I signed it... |
CLAs look good, thanks! ℹ️ Googlers: Go here for more info. |
Thank you for the change, I think it's correct but let me follow up with more knowledgeable folks in the TF team before I approve it. |
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.
This change is correct. Bummer.
@gabriel-vanzandycke can you also correct the converter tool to add the data_format arg? It should be parallel to the entry for erosion2d: https://github.com/tensorflow/tensorflow/blob/master/tensorflow/tools/compatibility/tf_upgrade_v2.py#L1525 @tomerk The converter didn't do the right thing here, can you review the change to make sure it will once @gabriel-vanzandycke is done? @tensorflow/api-owners This is a bug affecting the API: the accepted string has changed. The old state is a bug, we need to fix it. I added relnotes. |
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 add the converter code, see other 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.
Let's get this in, I'll submit the converter change separately.
…e-patch-dilation2d PiperOrigin-RevId: 261198637
gen_nn_ops.dilation2d
seems to be inNHWC
while the parent function was asking forNCHW
.I corrected the doc and the check.
RELNOTES: tf.nn.dilation2d now correctly requires its data_format argument to be "NHWC".