Home
last modified time | relevance | path

Searched refs:parameters_to_donate (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/pjrt/
Dutils.cc225 absl::flat_hash_set<int> parameters_to_donate; in GetParametersThatMustBeDonated() local
246 parameters_to_donate.insert(this_parameter); in GetParametersThatMustBeDonated()
256 parameters_to_donate.insert(this_parameter); in GetParametersThatMustBeDonated()
260 return parameters_to_donate; in GetParametersThatMustBeDonated()
Dpjrt_stream_executor_client.cc1611 TF_ASSIGN_OR_RETURN(absl::flat_hash_set<int> parameters_to_donate, in SetUpDonation()
1615 std::move(parameters_to_donate)); in SetUpDonation()