BUGFIX: Convert inputs and list of gradients into tuple if they are not...
BUGFIX: Convert inputs and list of gradients into tuple if they are not instance of tuple. Otherwise this causes "unhashable keys" error when we try to hash. Also fixed lint error. PiperOrigin-RevId: 195061425
Loading
Please sign in to comment