Avoid custom_gradient warnings

Open tk2lab opened this issue 4 years ago • 1 comments

python.interlnal.custom_gradient.custom_gradient generate warnings: "@custom_gradient grad_fn has 'variables' in signature, but no ResourceVariables were used on the forward pass.".

TF will specify not only arg 'variabels' but also **kwargs. In 'f_wrapped', vjp_bwd can use kwargs even if vjp_bwd_wrapped does not have kwargs.

Feb 14 '22 00:02 tk2lab

signed

Apr 27 '22 09:04 tk2lab