Searched refs:override_fn_when_gpu_float64 (Results 1 – 1 of 1) sorted by relevance
463 override_fn_when_gpu_float64=None, argument505 override_fn_when_gpu_float64509 return override_fn_when_gpu_float64(*[load(index) for load in loaders])720 …override_fn_when_gpu_float64=fn_libdevice if use_libdevice_for_f64 else None, # type: ignore[poss…