search

tensorflow / java

Java bindings for TensorFlow
Apache License 2.0
813 stars 200 forks source link

Fix custom gradient lookup bug #512

Closed karllessard closed 8 months ago

karllessard commented 8 months ago

The original code was doing the lookup on the node name, not the node type.