xla

cc3d82dc - Move add_step_closure out from build_graph