fuse_consecutive_squeezes issue: consecutive negtive axes
Nobody has claimed this yet.
- Dominant language
- C++
- Stars
- 834
- Forks
- 109
- Avg merge
- 6h 55m
- Merged PRs (30d)
- 2
Description
issue：
// Before:
// X is a tensor with shape=[1, 5000, 1, 1]
// Y = Squeeze(X, axes=[-1]) -> shape=[1, 5000, 1]
// Z = Squeeze(Y, axes=[-1]) -> shape=[1, 5000]
// After:
// Z = Squeeze(X, axes=[-1, 0]) -> shape=[5000, 1]
it should be
Z = Squeeze(X, axes=[-2, -1]) -> shape=[1, 5000]
or
Z = Squeeze(X, axes=[2, 3]) -> shape=[1, 5000]
codes resulted in line 93:
ret.push_back(i + prev_num + 1);
it's only right for positive axes values
Contributor guide
No contributing guide indexed for this repository
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.
Research direction
Start at line 93 in the fuse_consecutive_squeezes implementation and trace how consecutive negative axes are adjusted. Reproduce the issue with the tensor shapes and Squeeze examples in this report; done means the fused operation preserves the expected shape for negative and positive axes.
Written by the indexing model from the issue text.
Assessment
- Tech stack
- cpp
- Domain
- performance
- Issue type
- Bug
- Difficulty
- 3/5
- Estimated time
- 1-2 days
- Activity status
- Stale
- Clarity
- Mostly clear
- Newbie friendliness
- 35/100