Searched refs:_xla_call_transpose_update_params (Results 1 – 1 of 1) sorted by relevance
887 def _xla_call_transpose_update_params(params, undef_primals, nonzero_cts): function892 ad.call_transpose_param_updaters[xla_call_p] = _xla_call_transpose_update_params