Graph custom gradient support - #292

Merged

karllessard merged 27 commits into

tensorflow:masterfrom

rnett:rn_custom_gradients

Nov 11, 2021

Commits on Oct 20, 2021

Add JavaCPP generation for gradient registry stuff
rnett
committed
ee96979
Working gradients
rnett
committed
c5717b0
Add missing requireHandles
rnett
committed
12d579c
Rebase, use try-with-resources
rnett
committed
1f1ba2c
Expose the necessary symbols on Windows
rnett
committed
89690ab
Nicely handle pre-existing gradients
rnett
committed
dfaac8d
Small-ish review changes
rnett
committed
2f22b57
Use annotation instead of field reflection to store op types
rnett
committed
3acc001
Cleanup and more review changes
rnett
committed
5127413

Commits on Oct 29, 2021

Remove empty init file
rnett
committed
293fecc
Update annotation generator comments
rnett
committed
2eb9342

Commits on Oct 31, 2021

Update annotation names and comments, and registerCustomGradient javadoc
rnett
committed
f1f6e8b

Commits on Nov 5, 2021

Add no-arg ctor to BaseGradientAdapter
rnett
committed
8c40b8c
Add documentation about dangerousGradientBuilder
rnett
committed
d08b38b
Add Javadoc for getUnsafeNativeHandle
rnett
committed
72ed4f0
More dangerous gradient builder javadocs
rnett
committed
fd2609d
Add note about why gradientFuncs is required
rnett
committed
36a6e30
Store and allow getting native scope device when it has been set from Java
rnett
committed
2ddbb6c
Rename withDevice's parameter
rnett
committed
759a754
Update scope for fix review comments
rnett
committed
ed1da29
Clarify the difference between CustomGradient and RawCustomGradient
rnett
committed
4504a6f
Remove experiment
rnett
committed
be4840c

Commits on Nov 8, 2021

Adjust GraphOperation#input to not require a graph lock
rnett
committed
9601138
Remove printing from CustomGradientTest
rnett
committed
ca5d343
Cleanup adapter exceptions, name gradient scopes
rnett
committed
517fd8d

Commits on Nov 11, 2021

Document use of rawtypes
rnett
committed
f5dd0e5
Generate the new op classes
rnett
committed
bfbfb49