Home
last modified time | relevance | path

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

/dports/math/py-jax/jax-0.2.9/jax/interpreters/
H A Dad.py663 def _perm(primal_counts, tangent_counts, lst): argument
664 n = sum(primal_counts)
666 primal_groups = split_list(primals, primal_counts[:-1])