Searched defs:_RendezvousContext (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/distributed/elastic/rendezvous/ |
| H A D | dynamic_rendezvous.py | 543 class _RendezvousContext: class
574 state_handler: Callable[[_RendezvousContext, float], _Action],
638 state_handler: Callable[[_RendezvousContext, float], _Action],
844 def _should_keep_alive(ctx: _RendezvousContext) -> bool:
857 def __call__(self, ctx: _RendezvousContext, deadline: float) -> _Action:
868 def __call__(self, ctx: _RendezvousContext, deadline: float) -> _Action:
971 def __call__(self, ctx: _RendezvousContext, deadline: float) -> _Action:
982 def __call__(self, ctx: _RendezvousContext, deadline: float) -> _Action:
|