TFLite: Add fused_activation_function for ADD, SUB, MUL, DIV #3372
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.
I noticed that
ADD
,SUB
,MUL
andDIV
operators havefused_activation_function
option.https://raw.githubusercontent.com/tensorflow/tensorflow/r1.13/tensorflow/lite/schema/schema.fbs
This PR adds code to handle
fused_activation_function
option in_convert_elemwise
method.I did not modify any existing tests for these 4 operators because other operators (e.g.
FULLY_CONNECTED
orMAX_POOL_2D
withfused_activation_function
option do not have any tests withRELU
andRELU6
.