xla

Wrap def_static to enable warning reporting in static python functions.

#9380

Merged