Searched refs:fully_shard_with_compiled_compute (Results 1 – 1 of 1) sorted by relevance
1450 def fully_shard_with_compiled_compute(*args, **kwargs): function1479 … fully_shard_patch = fully_shard_with_compiled_compute # type: ignore[assignment]