(device_name_or_function)
| 5196 | elif executing_eagerly_outside_functions(): |
| 5197 | @tf_contextlib.contextmanager |
| 5198 | def combined(device_name_or_function): |
| 5199 | with get_default_graph().device(device_name_or_function): |
| 5200 | if not callable(device_name_or_function): |
| 5201 | with context.device(device_name_or_function): |
| 5202 | yield |
| 5203 | else: |
| 5204 | yield |
| 5205 | return combined(device_name_or_function) |
| 5206 | else: |
| 5207 | return get_default_graph().device(device_name_or_function) |
no test coverage detected