Home
last modified time | relevance | path

Searched refs:_notuple_allreduce_translation_rule (Results 1 – 1 of 1) sorted by relevance

/dports/math/py-jax/jax-0.2.9/jax/_src/lax/
H A Dparallel.py528 return _notuple_allreduce_translation_rule(
568 def _notuple_allreduce_translation_rule(prim, c, *args, axis_name, axis_env, function