Searched refs:undef_primals (Results 1 – 1 of 1) sorted by relevance
887 def _xla_call_transpose_update_params(params, undef_primals, nonzero_cts): argument889 donated_primals = [d for d, u in zip(donated_invars, undef_primals) if not u]