Searched refs:fft_p (Results 1 – 6 of 6) sorted by relevance
| /dports/math/py-jax/jax-0.2.9/jax/_src/lax/ |
| H A D | fft.py | 56 return fft_p.bind(x, fft_type=fft_type, fft_lengths=fft_lengths)
59 return xla.apply_primitive(fft_p, x, fft_type=fft_type, fft_lengths=fft_lengths)
134 fft_p = Primitive('fft') variable
135 fft_p.def_impl(fft_impl)
136 fft_p.def_abstract_eval(fft_abstract_eval)
137 xla.translations[fft_p] = fft_translation_rule
138 ad.deflinear2(fft_p, fft_transpose_rule)
139 batching.primitive_batchers[fft_p] = fft_batching_rule
141 xla.backend_specific_translations['cpu'][fft_p] = pocketfft.pocketfft
|
| /dports/audio/eq10q-lv2/eq10q-2.2/ |
| H A D | eq.c | 96 fftw_plan fft_p, fft_p2; member
118 fftw_destroy_plan(plugin->fft_p); in cleanupEQ()
277 …plugin_data->fft_p = fftw_plan_r2r_1d(FFT_N, plugin_data->fft_in, plugin_data->fft_out, FFTW_R2HC,… in instantiateEQ()
453 fftw_execute(plugin_data->fft_p); in runEQ_v2()
|
| /dports/math/py-jax/jax-0.2.9/jax/lax/ |
| H A D | __init__.py | 326 fft_p,
|
| /dports/math/py-jax/jax-0.2.9/jax/experimental/ |
| H A D | jet.py | 244 deflinear(lax_fft.fft_p)
|
| /dports/math/py-jax/jax-0.2.9/jax/experimental/jax2tf/tests/ |
| H A D | primitive_harness.py | 1531 lax.fft_p,
1533 lambda *args: lax.fft_p.bind(
|
| /dports/math/py-jax/jax-0.2.9/jax/experimental/jax2tf/ |
| H A D | jax2tf.py | 2078 tf_impl[lax_fft.fft_p] = _fft
|