Commit 4723f8f6 authored by RJ Ryan's avatar RJ Ryan Committed by TensorFlower Gardener
Browse files

Support SymbolicGradient for functions with non-trainable arguments. 

The non-trainable arguments end up with None as their incoming out_grad, which is not a valid input to SymbolicGradient (inputs have to be convertible to Tensor, and None isn't).

PiperOrigin-RevId: 173901727
parent 49467247
Show whitespace changes
Inline Side-by-side
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment