()
| 158 | |
| 159 | |
| 160 | def _enclosing_tpu_context(): |
| 161 | # pylint: disable=protected-access |
| 162 | context = ops.get_default_graph()._get_control_flow_context() |
| 163 | # pylint: enable=protected-access |
| 164 | while context is not None and not isinstance( |
| 165 | context, control_flow_ops.XLAControlFlowContext): |
| 166 | context = context.outer_context |
| 167 | return context |
| 168 | |
| 169 | |
| 170 | @tf_export('einsum', 'linalg.einsum') |
no test coverage detected