Adds support for Call-Context Arguments #843
Merged
Conversation
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
![copybara-service[bot]](https://avatars.githubusercontent.com/in/44061?s=80&v=4){kind=small}
copybara-service
bot
force-pushed
the
test_754990602
branch
from
87fcfea to
57dd4b4
Compare
Create an argument propagation flow for call-context arguments. Currently, keras uses `training` argument to infer whether layer should be called in training/inference mode. This change introduces a general flow of propagating arguments from a parent call to a child call (using call_context), so that we can add new control flow arguments in the future using a generic framework. This change does the following things: * Adds a `call_context_args` dictionary in the call_context object to store call-context arguments being propagated. * Changes current layer implementation to use the general propagation flow instead of using hardcoded `training`. * Adds utilities to query and set this context arguments in the `Layer` class. PiperOrigin-RevId: 761325027
copybara-service
bot
force-pushed
the
test_754990602
branch
from
57dd4b4 to
803835f
Compare
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
2 participants
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
Adds support for Call-Context Arguments
Create an argument propagation flow for call-context arguments.
Currently, keras uses
trainingargument to infer whether layer should be called in training/inference mode. This change introduces a general flow of propagating arguments from a parent call to a child call (using call_context), so that we can add new control flow arguments in the future using a generic framework.This change does the following things:
call_context_argsdictionary in the call_context object to store call-context arguments being propagated.training.Layerclass.