Commit 297fafbe authored by Yong Tang's avatar Yong Tang
Browse files

Support gradient_multipliers as tensor for optimize_loss



This fix tries to address the issue raised in 22295 where
gradient_multipliers for tf.contrib.layers.optimize_loss()
does not support tensor as input. This fix update the
optimize_loss to allow gradient_multipliers passed as dict
of tensors.

This fix fixes 22295.

Signed-off-by: default avatarYong Tang <yong.tang.github@outlook.com>
parent 95338704
Loading
Loading
Loading
Loading
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment