Home
last modified time | relevance | path

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

/dports/math/py-jax/jax-0.2.9/jax/experimental/
H A Djet.py218 def deflinear(prim): function
226 deflinear(lax.neg_p)
227 deflinear(lax.real_p)
228 deflinear(lax.complex_p)
229 deflinear(lax.conj_p)
230 deflinear(lax.imag_p)
231 deflinear(lax.add_p)
233 deflinear(lax.sub_p)
237 deflinear(lax.pad_p)
239 deflinear(lax.rev_p)
[all …]
/dports/math/py-jax/jax-0.2.9/jax/interpreters/
H A Dad.py427 def deflinear(primitive, transpose_rule): function