Fix control_flow_ops import (#3948)
Fix control_flow_ops import (#3948)

* Fix control_flow_ops import

Old access was not working on new version of tensorflow. This should
work for all versions.

* Fix identation

keras-team/keras,keras-team/keras,dolaameng/keras