Returns the current DistributionStrategy if in a cross-tower context.
Note that execution:
- starts in the default (single-tower) tower context;
- switches to cross-tower context when entering a
- switches to a (non-default) tower context inside
get_tower_context()->merge_call(merge_fn, ...), then inside
merge_fnyou are back in the cross-tower context.
Note that you can also go directly from step 1 to 4 to switch to a
cross-tower context for the default
DistributionStrategy. You may
also switch from the cross-tower context of 4 to a tower context by
call_for_each_tower(), jumping back to step 3.
DistributionStrategy methods may only be executed in
a cross-tower context.
Returns the current
DistributionStrategy object in a cross-tower
context, or None.
Exactly one of
will return None in a particular block.