Bug fix: consult graph's op registry to look up ops.
This is needed when the graph contains custom call ops. These functions are found only in the graph's registry and not the default one. PiperOrigin-RevId: 212297305
Loading
Please sign in to comment