Home
last modified time | relevance | path

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

/dports/math/py-jax/jax-0.2.9/jax/_src/lax/
H A Dfft.py59 return xla.apply_primitive(fft_p, x, fft_type=fft_type, fft_lengths=fft_lengths)
H A Dlinalg.py354 xla.apply_primitive(eig_p, operand,
451 v, w = xla.apply_primitive(eigh_p, operand, lower=lower)
808 lu, pivot, perm = xla.apply_primitive(lu_p, operand)
1021 q, r = xla.apply_primitive(qr_p, operand, full_matrices=full_matrices)
1125 return xla.apply_primitive(svd_p, operand, full_matrices=full_matrices,
H A Dlax.py1976 prim.def_impl(partial(xla.apply_primitive, prim))
3278 return xla.apply_primitive(broadcast_in_dim_p, operand, shape=shape,
3584 return xla.apply_primitive(reshape_p, operand, new_sizes=new_sizes,
5788 sort_p.def_impl(partial(xla.apply_primitive, sort_p))
5851 top_k_p.def_impl(partial(xla.apply_primitive, top_k_p))
5888 create_token_p.def_impl(partial(xla.apply_primitive, create_token_p))
5908 after_all_p.def_impl(partial(xla.apply_primitive, after_all_p))
5959 infeed_p.def_impl(partial(xla.apply_primitive, infeed_p))
5982 outfeed_p.def_impl(partial(xla.apply_primitive, outfeed_p))
6014 rng_uniform_p.def_impl(partial(xla.apply_primitive, rng_uniform_p))
[all …]
H A Dcontrol_flow.py541 while_p.def_impl(partial(xla.apply_primitive, while_p))
1104 cond_p.def_impl(partial(xla.apply_primitive, cond_p))
/dports/editors/poke/poke-1.0/jitter/example-vms/jitterlisp/
H A Djitterlisp-primitives.c630 JITTERLISP_PRIMITIVE_FUNCTION_2_(apply_primitive, PRIMITIVE, LIST,
895 apply_primitive),
/dports/lang/mit-scheme/mit-scheme-9.2/src/microcode/cmpauxmd/
H A Di386-nt.asm487 ;;; apply_primitive
H A Di386-ntw.asm487 ;;; apply_primitive
H A Di386.m4783 ### apply_primitive
/dports/math/py-jax/jax-0.2.9/jax/experimental/
H A Dhost_callback.py777 return xla.apply_primitive(outside_call_p, *args, **params)
/dports/math/py-jax/jax-0.2.9/jax/_src/
H A Drandom.py207 threefry2x32_p.def_impl(partial(xla.apply_primitive, threefry2x32_p))
/dports/math/py-jax/jax-0.2.9/jax/interpreters/
H A Dxla.py236 def apply_primitive(prim, *args, **params): function