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 D | fft.py | 59 return xla.apply_primitive(fft_p, x, fft_type=fft_type, fft_lengths=fft_lengths)
|
H A D | linalg.py | 354 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 D | lax.py | 1976 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 D | control_flow.py | 541 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 D | jitterlisp-primitives.c | 630 JITTERLISP_PRIMITIVE_FUNCTION_2_(apply_primitive, PRIMITIVE, LIST, 895 apply_primitive),
|
/dports/lang/mit-scheme/mit-scheme-9.2/src/microcode/cmpauxmd/ |
H A D | i386-nt.asm | 487 ;;; apply_primitive
|
H A D | i386-ntw.asm | 487 ;;; apply_primitive
|
H A D | i386.m4 | 783 ### apply_primitive
|
/dports/math/py-jax/jax-0.2.9/jax/experimental/ |
H A D | host_callback.py | 777 return xla.apply_primitive(outside_call_p, *args, **params)
|
/dports/math/py-jax/jax-0.2.9/jax/_src/ |
H A D | random.py | 207 threefry2x32_p.def_impl(partial(xla.apply_primitive, threefry2x32_p))
|
/dports/math/py-jax/jax-0.2.9/jax/interpreters/ |
H A D | xla.py | 236 def apply_primitive(prim, *args, **params): function
|