TAKEKAWA Takashi

Avoid custom_gradient warnings

1

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',...