Skip to content

Allow excluding Softmax from XLA clustering#105

Draft
stevenvar wants to merge 1 commit into
joeyye-work:for-serving-2.20from
stevenvar:steven.xla_exclude_softmax
Draft

Allow excluding Softmax from XLA clustering#105
stevenvar wants to merge 1 commit into
joeyye-work:for-serving-2.20from
stevenvar:steven.xla_exclude_softmax

Conversation

@stevenvar
Copy link
Copy Markdown
Collaborator

@stevenvar stevenvar commented May 5, 2026

Adds Softmax support to the existing tf_xla_cluster_exclude_ops mechanism.

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.

1 participant