Is there any way to implement nonzero and unique through the tensorrt plugin?
Open
Nobody has claimed this yet.
triaged
- Dominant language
- C++
- Stars
- 13.4k
- Forks
- 2.4k
- Avg merge
- 5d 3h
- Merged PRs (30d)
- 2
Description
I tried to do this with the tensorrt plugin, but you need to specify the output dimensions, and the output dimensions of these two operators are dynamic and determined in the enqueue.
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.
Research direction
Start by reviewing the TensorRT plugin interface for output-dimension handling and the enqueue stage. Determine whether nonzero and unique can support dimensions computed during enqueue, and document the supported approach or limitation; the issue names no files or tests to run.
Written by the indexing model from the issue text.
Assessment
- Tech stack
- cpp
- Domain
- machine-learning
- Issue type
- Feature
- Difficulty
- 5/5
- Estimated time
- Over a week
- Activity status
- Stale
- Clarity
- Needs clarification
- Newbie friendliness
- 25/100