Skip to content

[TF] Add DenseBincount support#12728

Merged
masahi merged 1 commit intoapache:mainfrom
apivovarov:tf_bincount
Sep 9, 2022
Merged

[TF] Add DenseBincount support#12728
masahi merged 1 commit intoapache:mainfrom
apivovarov:tf_bincount

Conversation

@apivovarov
Copy link
Contributor

@apivovarov apivovarov commented Sep 8, 2022

Description

This PR adds TF DenseBincount support. The code was inspired by TVM pytorch.py bincount impl.

Limitations:

  • Requires relay.vm.compile because internal data structures shapes depend on the input data content.

Known issues:

  • Input array should not contain numbers outside of the range [0, size-1]. TVM scatter_add counts such values, but TF DenseBincount ignores them.

@masahi

@masahi masahi merged commit cb08a12 into apache:main Sep 9, 2022
apivovarov added a commit to neo-ai/tvm that referenced this pull request Sep 13, 2022
apivovarov added a commit to neo-ai/tvm that referenced this pull request Nov 3, 2022
apivovarov added a commit to neo-ai/tvm that referenced this pull request Nov 18, 2022
xinetzone pushed a commit to daobook/tvm that referenced this pull request Nov 25, 2022
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants