Searched defs:dtypesIfMPS (Results 1 – 2 of 2) sorted by relevance
1555 class dtypesIfMPS(dtypes): class
218 …dtypesIfMPS=(torch.float16, torch.float32,), # dtypes this function is expected to work with on M… argument