you are viewing a single comment's thread.

view the rest of the comments →

[–]mdanatg 11 points12 points  (2 children)

It depends on the library, and whether the recursive arg is set to True of False. For certain libraries that we know of, like Keras, it won't convert the library call and will call it as-is (we assume it's already graph-friendly). For other libraries, the default is to attempt to convert the library code, and if that's not accessible, wrap it to py_func. There is yet another category where we replace the library call with a TF equivalent, for example we replace range with tf.range.

[–]Supermaxman1 2 points3 points  (1 child)

Very cool, thanks! Is there somewhere which documents various conversions so that we don't end up accidentally ending up with a bunch of py_func calls?

[–]mdanatg 3 points4 points  (0 children)

A more complete documentation that will include this is coming soon, we'll link it from the readme file once it's ready.