Custom/gradients dispatch - #632

Merged

Craigacp merged 12 commits into

tensorflow:masterfrom

nfeybesse:custom/gradients-dispatch

Mar 9, 2026

Commits on Feb 8, 2026

Add GradientDispatch bridge for custom gradient adapter dispatch
nfeybesse
committed
bcf9e2e
Fix custom gradient registration scalability and support NoGradient
nfeybesse
committed
7c7dc54
Handle NoGradient in native custom gradient bridge
nfeybesse
committed
e2fa04b

Commits on Feb 10, 2026

Add tests for NoGradient support in Java custom gradients
nfeybesse
committed
1d43d4c
Fix custom gradients: support NoGradient and stabilize adapter
nfeybesse
committed
69a9a36

Commits on Feb 11, 2026

apply mvn spotless
nfeybesse
committed
8d80312

Commits on Feb 12, 2026

Fix review comments: enforce mutual exclusion for raw/typed gradients, remove inline ifs, add license headers and imports
nfeybesse
committed
1240293

Commits on Feb 13, 2026

Add Javadoc to DispatchingGradientAdapter
nfeybesse
committed
fd1cc19

Commits on Feb 14, 2026

Add Apache 2.0 license header to GradientTests
nfeybesse
committed
cdd943c

Commits on Feb 24, 2026

Add test registering >10 custom gradients and verifying registry presence
nfeybesse
committed
56539b9

Commits on Feb 25, 2026

Use single native dispatch adapter for custom gradients
nfeybesse
committed
6be7c17

Commits on Mar 7, 2026

Minimising the custom gradient fix.

Craigacp
authored and
nfeybesse
committed
da6c216