Skip to content
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

Utilize TensorFormat #42102

Merged

Conversation

WindQAQ
Copy link
Member

@WindQAQ WindQAQ commented Aug 6, 2020

No description provided.

@google-ml-butler google-ml-butler bot added the size:S CL Change Size: Small label Aug 6, 2020
@gbaned gbaned self-assigned this Aug 6, 2020
@gbaned gbaned added this to Assigned Reviewer in PR Queue via automation Aug 6, 2020
@WindQAQ
Copy link
Member Author

WindQAQ commented Aug 6, 2020

/cc @smit-hinsu for visibility.

Copy link
Contributor

@smit-hinsu smit-hinsu left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thanks for the change!

@@ -339,15 +339,18 @@ void BatchToSpaceOp::getCanonicalizationPatterns(
// are not unknown.
//
static LogicalResult Verify(BiasAddOp op) {
StringRef format = op.data_format();
if (format == "NHWC") {
std::string data_format = op.data_format().str();
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

optional nit: To avoid materializing the string here for FormatFromString, let's change input type of the function to be string_view. Then, we can convert from StringRef to string_view directly.

PR Queue automation moved this from Assigned Reviewer to Approved by Reviewer Aug 6, 2020
@google-ml-butler google-ml-butler bot added kokoro:force-run Tests on submitted change ready to pull PR ready for merge process labels Aug 6, 2020
@kokoro-team kokoro-team removed kokoro:force-run Tests on submitted change labels Aug 6, 2020
@tensorflow-copybara tensorflow-copybara merged commit c415c83 into tensorflow:master Aug 7, 2020
PR Queue automation moved this from Approved by Reviewer to Merged Aug 7, 2020
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
cla: yes ready to pull PR ready for merge process size:S CL Change Size: Small
Projects
PR Queue
  
Merged
Development

Successfully merging this pull request may close these issues.

None yet

6 participants