Searched refs:_raise_exception_fn (Results 1 – 1 of 1) sorted by relevance
74 def _raise_exception_fn(_=None): function81 ds_context.get_replica_context().merge_call(_raise_exception_fn)88 dist.extended.call_for_each_replica(_raise_exception_fn)304 dist.extended.call_for_each_replica(_raise_exception_fn)