Repository navigation

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