(state_dict: dict[str, torch.Tensor], dtype: torch.dtype = torch.float16)
source not stored for this graph (policy: none)
nothing calls this directly
no test coverage detected