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
Extract a function for parsing op DEPTH_TO_SPACE #46026
Extract a function for parsing op DEPTH_TO_SPACE #46026
Conversation
Extract the parsing of op parameters from the flatbuffer out of a switch statement case, into a standalone function which can be called by the micro op resolver.
@@ -1059,7 +1059,7 @@ TfLiteStatus ParseDepthToSpace(const Operator* op, | |||
if (schema_params != nullptr) { | |||
params->block_size = schema_params->block_size(); | |||
} else { | |||
// TODO(b/157480169): We should either return kTfLiteError or fill in some | |||
// old todo (b/157480169): We should either return kTfLiteError or fill in some |
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.
we do want to keep this TODO as is. If it is triggering some failure in the CI then we'll need to fix the CI. Can you please change it back to TODO(b/): and then I'll take a look at what the CI error is.
All checks are passing. Merging a PR needs two approvals from a Googler so we'll need to wait until @petewarden can approve the change in the internal review system before this can be merged. |
Merge branch 'master' of https://github.com/tensorflow/tensorflow into feature-micro-add-op-depth-to-space-pr1
@advaitjain not sure what internal check this one is failing. |
The failing tests are internal, unrelated and currently flaky. I have set the internal change to submit while ignoring these tests. This PR should get merged shortly. |
Extract the parsing of op parameters from the flatbuffer out of a
switch statement case, into a standalone function which can be
called by the micro op resolver.
This PR is part of the work to port operator DEPTH_TO_SPACE
from lite to micro, as tracked in issue #46025.