Searched refs:_xla_call_jvp_update_params (Results 1 – 1 of 1) sorted by relevance
880 def _xla_call_jvp_update_params(params, nz_tangents): function885 ad.call_param_updaters[xla_call_p] = _xla_call_jvp_update_params