Searched refs:get_device_states (Results 1 – 2 of 2) sorted by relevance
166 def get_device_states(*args) -> Tuple[List[int], List[torch.Tensor]]: function246 ctx.fwd_devices, ctx.fwd_device_states = get_device_states(*args)1498 fwd_devices, fwd_device_states = get_device_states(*args)
48 get_device_states,509 device_ids, device_states = get_device_states(inp)