Commit 40d4a315 authored by Asim Shankar's avatar Asim Shankar Committed by TensorFlower Gardener
Browse files

eager: gradients_function now accepts primitives and numpy arrays. 

Fixed bug which changed the output type of the gradients_function
to be a list of primitives instead of a list of Tensors if the "dy"
argument was provided.

PiperOrigin-RevId: 168748297
parent 646ccee7
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