tensorflow / tensorflow/model-optimization
Support for Multiply layer
@Xhark is already working on this.
Since Jun 15, 2021.
- Dominant language
- Python
- Stars
- 1.6k
- Forks
- 349
- Avg merge
- 3d 2h
- Merged PRs (30d)
- 1
Description
System information
- TensorFlow version (you are using): 2.4.1
- Are you willing to contribute it (Yes/No): no
Motivation
The implementation of some models (e.g. SENet) requires the use of the Multiply layer
Describe the feature
I tried to quantize a model including a Squeeze-Excitation block but I got an error:
Layer multiply:<class 'tensorflow.python.keras.layers.merge.Multiply'> is not supported. You can quantize this layer by passing a tfmot.quantization.keras.QuantizeConfig instance to the quantize_annotate_layer API.
It would be very useful to have this layer supported
Describe how the feature helps achieve the use case
It would be possible to fully quantize models including this layer
Contributor guide
First steps
- Read the whole issue, then the project's contributing guide.
- Comment on the issue to say you are picking it up — it saves two people doing the same work.
- Fork the repository and make your change on a branch.
- Open a pull request that references the issue number.
Assessment
This issue has not been assessed yet.