Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/pjrt/
Dpjrt_stream_executor_client.h714 std::vector<absl::flat_hash_set<int>> parameters_that_must_be_donated_; variable
Dpjrt_stream_executor_client.cc1609 parameters_that_must_be_donated_.reserve(executables_.size()); in SetUpDonation()
1614 parameters_that_must_be_donated_.emplace_back( in SetUpDonation()
1631 return parameters_that_must_be_donated_[executable_idx].contains(parameter); in MustDonateParameter()