markdaoust:

But have you seen the experimental_get_compiler_ir method on the GenericFunction class returned by tf.function? Is there maybe a way that it could help here?


I think this is at runtime jit and not for aot, right?
If you know someone that is working on this area It could be nice to know how users could use this with bazel build and add the command in that Doc page.