alibaba / alibaba/TinyNeuralNetwork
Could you please support the conversion of "grid_sample" operator?
Open
enhancement
work/x-large
- Dominant language
- Python
- Stars
- 879
- Forks
- 134
- PR merge metrics
- No merged PRs in 30d
Description
The pytorch "grid_sample" operator can be translated to tensorflow "gather_nd", can you please support this conversion?
reference: https://github.com/hellochick/PWCNet-tf2/blob/master/flow_utils.py, bilinear_warp function
Contributor guide
Research direction
Start by reading the linked flow_utils.py bilinear_warp function and its gather_nd-based handling. Then inspect TinyNeuralNetwork's existing PyTorch-to-TensorFlow operator conversion entry points and tests, if present. Done means grid_sample models convert successfully with equivalent behavior while existing conversions remain unaffected.
Written by the indexing model from the issue text.
Assessment
- Tech stack
- python, pytorch, tensorflow
- Domain
- machine-learning
- Issue type
- Feature
- Difficulty
- 4/5
- Estimated time
- 3-5 days
- Activity status
- Stale
- Clarity
- Mostly clear
- Newbie friendliness
- 35/100