Searched refs:patch_unshard (Results 1 – 3 of 3) sorted by relevance
48 patch_unshard,461 with patch_unshard(unshard_with_record), patch_post_backward(512 with patch_unshard(unshard_with_record), patch_post_backward(589 with patch_unshard(unshard_with_record), patch_post_backward(675 with patch_unshard(unshard_with_record), patch_reshard(760 with patch_unshard(unshard_with_record), patch_reshard(840 with patch_unshard(unshard_with_record), patch_post_backward(919 with patch_unshard(unshard_with_record), patch_post_backward(
208 def patch_unshard(_patched_unshard: Callable): function235 with patch_unshard(
989 def patch_unshard(new_unshard: Callable): function