Code

Graph custom gradient support#292

Code

Merged

karllessard merged 27 commits intotensorflow:mastertensorflow/java:masterfrom rnett:rn_custom_gradientsrnett/tensorflow-java:rn_custom_gradientsCopy head branch name to clipboardNov 11, 2021

+17,408-2,986

Commits

Commits on Oct 20, 2021

• Add JavaCPP generation for gradient registry stuff

  Show description for ee96979
  rnett

  committed
  ee96979

  Copy full SHA for ee96979

• Working gradients

  Show description for c5717b0
  rnett

  committed
  c5717b0

  Copy full SHA for c5717b0

• Add missing requireHandles

  Show description for 12d579c
  rnett

  committed
  12d579c

  Copy full SHA for 12d579c

• Rebase, use try-with-resources

  Show description for 1f1ba2c
  rnett

  committed
  1f1ba2c

  Copy full SHA for 1f1ba2c

• Expose the necessary symbols on Windows

  Show description for 89690ab
  rnett

  committed
  89690ab

  Copy full SHA for 89690ab

• Nicely handle pre-existing gradients

  Show description for dfaac8d
  rnett

  committed
  dfaac8d

  Copy full SHA for dfaac8d

• Small-ish review changes

  Show description for 2f22b57
  rnett

  committed
  2f22b57

  Copy full SHA for 2f22b57

• Use annotation instead of field reflection to store op types

  Show description for 3acc001
  rnett

  committed
  3acc001

  Copy full SHA for 3acc001

• Cleanup and more review changes

  Show description for 5127413
  rnett

  committed
  5127413

  Copy full SHA for 5127413

Commits on Oct 29, 2021

• Remove empty init file

  Show description for 293fecc
  rnett

  committed
  293fecc

  Copy full SHA for 293fecc

• Update annotation generator comments

  Show description for 2eb9342
  rnett

  committed
  2eb9342

  Copy full SHA for 2eb9342

Commits on Oct 31, 2021

• Update annotation names and comments, and registerCustomGradient javadoc

  Show description for f1f6e8b
  rnett

  committed
  f1f6e8b

  Copy full SHA for f1f6e8b

Commits on Nov 5, 2021

• Add no-arg ctor to BaseGradientAdapter

  Show description for 8c40b8c
  rnett

  committed
  8c40b8c

  Copy full SHA for 8c40b8c

• Add documentation about dangerousGradientBuilder

  Show description for d08b38b
  rnett

  committed
  d08b38b

  Copy full SHA for d08b38b

• Add Javadoc for getUnsafeNativeHandle

  Show description for 72ed4f0
  rnett

  committed
  72ed4f0

  Copy full SHA for 72ed4f0

• More dangerous gradient builder javadocs

  Show description for fd2609d
  rnett

  committed
  fd2609d

  Copy full SHA for fd2609d

• Add note about why gradientFuncs is required

  Show description for 36a6e30
  rnett

  committed
  36a6e30

  Copy full SHA for 36a6e30

• Store and allow getting native scope device when it has been set from Java

  Show description for 2ddbb6c
  rnett

  committed
  2ddbb6c

  Copy full SHA for 2ddbb6c

• Rename withDevice's parameter

  Show description for 759a754
  rnett

  committed
  759a754

  Copy full SHA for 759a754

• Update scope for fix review comments

  Show description for ed1da29
  rnett

  committed
  ed1da29

  Copy full SHA for ed1da29

• Clarify the difference between CustomGradient and RawCustomGradient

  Show description for 4504a6f
  rnett

  committed
  4504a6f

  Copy full SHA for 4504a6f

• Remove experiment

  Show description for be4840c
  rnett

  committed
  be4840c

  Copy full SHA for be4840c

Commits on Nov 8, 2021

• Adjust GraphOperation#input to not require a graph lock

  Show description for 9601138
  rnett

  committed
  9601138

  Copy full SHA for 9601138

• Remove printing from CustomGradientTest

  Show description for ca5d343
  rnett

  committed
  ca5d343

  Copy full SHA for ca5d343

• Cleanup adapter exceptions, name gradient scopes

  Show description for 517fd8d
  rnett

  committed
  517fd8d

  Copy full SHA for 517fd8d

Commits on Nov 11, 2021

• Document use of rawtypes

  Show description for f5dd0e5
  rnett

  committed
  f5dd0e5

  Copy full SHA for f5dd0e5

• Generate the new op classes

  Show description for bfbfb49
  rnett

  committed
  bfbfb49

  Copy full SHA for bfbfb49