sony / model_optimization

Model Compression Toolkit (MCT) is an open source project for neural network model optimization under efficient, constrained hardware. This project provides researchers, developers, and engineers advanced quantization and compression tools for deploying state-of-the-art neural networks.
https://sony.github.io/model_optimization/
Apache License 2.0
332 stars 53 forks source link

Support torch expand with const #1231

Closed elad-c closed 2 months ago

elad-c commented 2 months ago

Pull Request Description:

Add torch.Tensor.expand to TPC.v4. Fix const quantization for torch expand & gather.

Checklist before requesting a review: