()
| 27 | |
| 28 | |
| 29 | def GetDefaultSession(): |
| 30 | global _sess_id2sess |
| 31 | default_sess_id = oneflow._oneflow_internal.GetDefaultSessionId() |
| 32 | assert default_sess_id in _sess_id2sess |
| 33 | return _sess_id2sess[default_sess_id] |
| 34 | |
| 35 | |
| 36 | def NewDefaultSession(env): |