Searched refs:tuple_inputs (Results 1 – 4 of 4) sorted by relevance
210 const HloModule& module, bool tuple_inputs) { in GetParametersThatMustBeDonated() argument213 if (tuple_inputs) { in GetParametersThatMustBeDonated()230 if (tuple_inputs) { in GetParametersThatMustBeDonated()
53 const HloModule& module, bool tuple_inputs);
677 Status SetUpDonation(bool tuple_inputs);739 const HloModule& hlo_module, bool tuple_inputs);
1608 Status PjRtStreamExecutorExecutable::SetUpDonation(bool tuple_inputs) { in SetUpDonation() argument1613 executable->executable()->module(), tuple_inputs)); in SetUpDonation()