_batched_reduction_collective (reference) in projects: dports

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

/dports/math/py-jax/jax-0.2.9/jax/_src/lax/
H A D	parallel.py	`501 def _batched_reduction_collective( function 603 partial(_batched_reduction_collective, 631 partial(_batched_reduction_collective, pmax_p, 642 partial(_batched_reduction_collective, pmin_p,`