Searched refs:lower_fun_initial_style (Results 1 – 3 of 3) sorted by relevance
365 xla.lower_fun_initial_style(_custom_jvp_call_jaxpr_impl)665 xla.lower_fun_initial_style(_custom_vjp_call_jaxpr_impl)
1878 xla.initial_style_translations[scan_p] = xla.lower_fun_initial_style(_scan_impl)2358 xla.lower_fun_initial_style(_custom_linear_solve_impl)
980 def lower_fun_initial_style(fun): function